Searched defs:TensorSpec (Results 1 – 5 of 5) sorted by relevance
| /llvm-project/llvm/lib/Analysis/models/ |
| H A D | log_reader.py | 28 class TensorSpec: class
51 def __init__(self, spec: TensorSpec, buffer: bytes):
69 def read_tensor(fs: io.BufferedReader, ts: TensorSpec) -> TensorValue:
91 tensor_specs: List[TensorSpec],
92 score_spec: Optional[TensorSpec],
|
| H A D | interactive_host.py | 23 def send(f: io.BufferedWriter, value: Union[int, float], spec: log_reader.TensorSpec):
|
| /llvm-project/llvm/include/llvm/Analysis/ |
| H A D | TensorSpec.h | 94 TensorSpec(const std::string &NewName, const TensorSpec &Other) in TensorSpec() function
|
| /llvm-project/llvm/lib/Analysis/ |
| H A D | ModelUnderTrainingRunner.cpp | 60 if (auto TensorSpec = getTensorSpecFromJSON(Ctx, *SpecPart)) in loadOutputSpecs() local
|
| H A D | TensorSpec.cpp | 62 TensorSpec::TensorSpec(const std::string &Name, int Port, TensorType Type, in TensorSpec() function in llvm::TensorSpec
|