Searched defs:TensorSpec (Results 1 – 10 of 10) sorted by relevance
| /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() function
|
| /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() argument
|
| /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/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],
|