| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | shape_util.cc | 89 TensorShape* tensor_shape) { in XLAShapeToTensorShape()
104 const PartialTensorShape& tensor_shape, in TensorShapeToXLAShape()
113 const PartialTensorShape& tensor_shape) { in TensorShapeToXLAShape()
145 Status TensorShapeToXLAShape(DataType dtype, const TensorShape& tensor_shape, in TensorShapeToXLAShape()
154 const TensorShape& tensor_shape) { in TensorShapeToXLAShape()
|
| D | xla_op_kernel.cc | 279 TensorShape tensor_shape({}); in ResolveInputDynamismIntoPred() local
309 TensorShape tensor_shape({InputShape(index).num_elements()}); in ResolveInputDynamismIntoPredVector() local
560 TensorShape tensor_shape; in SetOutputExpression() local
|
| D | xla_compiler.cc | 762 TensorShape tensor_shape; in CompileFunction() local
772 TensorShape tensor_shape = absl::get<TensorShape>(args[i].shape); in CompileFunction() local
|
| /external/tensorflow/tensorflow/core/data/ |
| D | dataset.proto | 15 .tensorflow.TensorShapeProto tensor_shape = 2; field
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | fake_param_op.cc | 34 PartialTensorShape tensor_shape; in XlaFakeParamOp() local
|
| D | sendrecv_ops.cc | 68 TensorShape tensor_shape; in RecvOp() local
|
| /external/tensorflow/tensorflow/python/keras/protobuf/ |
| D | projector_config.proto | 36 repeated uint32 tensor_shape = 4; field
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | snapshot.proto | 37 .tensorflow.TensorShapeProto tensor_shape = 2; field
|
| D | meta_graph.proto | 257 TensorShapeProto tensor_shape = 3; field
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | list_ops.cc | 174 shape_inference::ShapeHandle tensor_shape = c->UnknownShape(); in __anonabedecb60502() local
328 shape_inference::ShapeHandle tensor_shape = c->input(0); in __anonabedecb60902() local
362 shape_inference::ShapeHandle tensor_shape = c->input(0); in __anonabedecb60a02() local
|
| D | data_flow_ops.cc | 756 ShapeHandle tensor_shape = shapes->at(0).shape; in __anonea01102b1902() local
781 ShapeHandle tensor_shape = shapes->at(0).shape; in __anonea01102b1a02() local
821 ShapeHandle tensor_shape = shapes->at(0).shape; in __anonea01102b1b02() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor.proto | 20 TensorShapeProto tensor_shape = 2; field
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/ |
| D | ragged_tensor_to_tensor_tflite.cc | 118 const RuntimeShape tensor_shape(tensor.dims->size, tensor.dims->data); in TensorShapeFromTensor() local
156 const RuntimeShape tensor_shape(tensor->dims->size, tensor->dims->data); in GetMaxWidthValueRowID() local
187 const RuntimeShape tensor_shape(tensor->dims->size, tensor->dims->data); in GetMaxWidthRowSplit() local
335 const RuntimeShape tensor_shape(value_rowids.dims->size, in CalculateOutputIndexValueRowID() local
|
| /external/tensorflow/tensorflow/lite/micro/ |
| D | test_helpers.cc | 356 const int32_t tensor_shape[tensor_shape_size] = {1}; in BuildSimpleMockModel() local
446 const int32_t tensor_shape[tensor_shape_size] = {1}; in BuildComplexMockModel() local
582 const int32_t tensor_shape[tensor_shape_size] = {1}; in BuildSimpleMultipleInputsModel() local
892 const int32_t tensor_shape[tensor_shape_size] = {size}; in Create1dFlatbufferTensor() local
915 const int32_t tensor_shape[tensor_shape_size] = {size}; in CreateQuantizedFlatbufferTensor() local
933 const int32_t tensor_shape[tensor_shape_size] = {size}; in CreateMissingQuantizationFlatbufferTensor() local
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_from_tensor_op_test.py | 545 def testPartialShapes(self, tensor, tensor_shape, shape=None, argument
|
| /external/tensorflow/tensorflow/core/profiler/protobuf/ |
| D | memory_profile.proto | 59 string tensor_shape = 9; field
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | load_and_remap_matrix_op.cc | 137 TensorShape tensor_shape; in Compute() local
|
| D | random_op.cc | 56 TensorShape tensor_shape; in AllocateOutputWithShape() local
|
| D | remote_fused_graph_execute_op_test.cc | 89 const TensorShape& tensor_shape, Scope* root) { in BuildPlaceHolderOp()
|
| D | sparse_conditional_accumulator.h | 89 const Tensor* tensor_shape = std::get<2>(*tensor); in ValidateShape() local
|
| D | parameterized_truncated_normal_op.cc | 641 TensorShape tensor_shape; in Compute() local
|
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_format.h | 468 inline int64 GetTensorDim(const TensorShape& tensor_shape, in GetTensorDim()
476 inline int64 GetFilterDim(const TensorShape& tensor_shape, in GetFilterDim()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | linalg_ops.py | 304 def _use_composite_impl(fast, tensor_shape): argument
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | costmodel.cc | 253 const TensorShapeProto& tensor_shape, in RecordMaxMemorySize()
557 Bytes CostModel::MinTensorMemoryUsage(const TensorShapeProto& tensor_shape, in MinTensorMemoryUsage()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | op_def_util.cc | 368 const TensorShapeProto& tensor_shape) { in TensorShapeProtoToPyObject()
|