| /external/tensorflow/tensorflow/cc/experimental/libtf/impl/ |
| D | tensor_spec.h | 30 struct TensorSpec { struct
34 bool operator==(const TensorSpec& o) const { argument
40 friend H AbslHashValue(H h, const TensorSpec& t) { in AbslHashValue() argument
|
| /external/tensorflow/tensorflow/core/tfrt/saved_model/ |
| D | saved_model.h | 52 struct TensorSpec { struct
56 explicit TensorSpec(tensorflow::DataType dtype) : dtype(dtype) {} in TensorSpec() argument
57 TensorSpec(tensorflow::DataType dtype, tensorflow::PartialTensorShape shape) in TensorSpec() argument
319 using TensorSpec = ::tensorflow::tfrt_stub::TensorSpec; variable
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ |
| D | tensor_spec.cc | 25 TensorSpec::TensorSpec() in TensorSpec() function in tensorflow::TensorSpec
28 TensorSpec::TensorSpec(PartialTensorShape shape, DataType dtype) in TensorSpec() function in tensorflow::TensorSpec
31 TensorSpec::TensorSpec(const TensorSpecProto& proto) in TensorSpec() function in tensorflow::TensorSpec
|
| /external/ComputeLibrary/src/core/NEON/kernels/arm_conv/pooling/ |
| D | depthfirst_driver.hpp | 47 struct TensorSpec struct
49 T base;
50 size_t ld_row, ld_col;
52 TensorSpec(T ptr, size_t ld_row, size_t ld_col) in TensorSpec() argument
|
| /external/swiftshader/third_party/llvm-16.0/llvm/lib/Analysis/models/ |
| D | log_reader.py | 27 class TensorSpec: class
50 def __init__(self, spec: TensorSpec, buffer: bytes):
69 def read_tensor(fs: typing.BinaryIO, ts: TensorSpec) -> TensorValue:
|
| /external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/ |
| D | depthfirst_driver.hpp | 55 struct TensorSpec struct
57 T base;
58 size_t ld_row, ld_col;
60 TensorSpec(T ptr, size_t ld_row, size_t ld_col) in TensorSpec() function
|
| /external/tensorflow/tensorflow/dtensor/python/ |
| D | input_util.py | 107 element_spec: tensor_spec.TensorSpec, layouts: Any,
211 flattened_elem_spec: Sequence[tensor_spec.TensorSpec],
292 elem_spec: tensor_spec.TensorSpec) -> ops.Tensor:
|
| /external/federated-compute/fcp/aggregation/core/ |
| D | tensor_spec.h | 32 TensorSpec(std::string name, DataType dtype, TensorShape shape) in TensorSpec() function
|
| /external/swiftshader/third_party/llvm-16.0/llvm/include/llvm/Analysis/ |
| D | TensorSpec.h | 86 TensorSpec(const std::string &NewName, const TensorSpec &Other) in TensorSpec() function
|
| /external/swiftshader/third_party/llvm-16.0/llvm/lib/Analysis/ |
| D | TensorSpec.cpp | 60 TensorSpec::TensorSpec(const std::string &Name, int Port, TensorType Type, in TensorSpec() function in llvm::TensorSpec
|
| D | ModelUnderTrainingRunner.cpp | 60 if (auto TensorSpec = getTensorSpecFromJSON(Ctx, *SpecPart)) in loadOutputSpecs() local
|
| /external/federated-compute/fcp/artifact_building/ |
| D | checkpoint_utils.py | 201 tensor_specs: Iterable[tf.TensorSpec], filepath_tensor: tf.Tensor
|
| D | federated_compute_plan_builder.py | 1176 broadcasted_tensor_specs: Iterable[tf.TensorSpec],
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | tensor_spec.py | 106 class TensorSpec(DenseSpec, type_spec.BatchableTypeSpec, class
|
| /external/tensorflow/tensorflow/python/ops/structured/ |
| D | structured_tensor.py | 1573 tensor_spec.TensorSpec]
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | dynamic_ragged_shape_test.py | 3106 a: tensor_spec.TensorSpec,
3107 b: tensor_spec.TensorSpec) -> None:
|
| D | dynamic_ragged_shape.py | 1310 tensor_spec.TensorSpec],
|