| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | shape_util.cc | 88 TensorShape* tensor_shape) { in XLAShapeToTensorShape()
102 Status TensorShapeToXLAShape(DataType dtype, const TensorShape& tensor_shape, in TensorShapeToXLAShape()
111 const TensorShape& tensor_shape) { in TensorShapeToXLAShape()
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/ |
| D | batch_features.cc | 98 auto tensor_shape = TensorShape({shape_flat(0), shape_flat(1)}); in Initialize() local
140 auto tensor_shape = TensorShape({shape_flat(0), shape_flat(1)}); in Initialize() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | fake_param_op.cc | 32 TensorShape tensor_shape; in XlaFakeParamOp() local
|
| D | sendrecv_ops.cc | 68 TensorShape tensor_shape; in RecvOp() local
|
| D | tensor_list_ops.cc | 276 const TensorShape tensor_shape = ctx->InputShape(0); in Compile() local
|
| /external/tensorflow/tensorflow/contrib/tensorboard/plugins/projector/ |
| D | projector_config.proto | 32 repeated uint32 tensor_shape = 4; field
|
| /external/tensorflow/tensorflow/contrib/mpi_collectives/ |
| D | mpi_message.proto | 39 TensorShapeProto tensor_shape = 5; field
|
| D | mpi_ops.cc | 340 TensorShape tensor_shape = requests[0].tensor_shape(); in ConstructMPIResponse() local
363 TensorShape tensor_shape = requests[0].tensor_shape(); in ConstructMPIResponse() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor.proto | 18 TensorShapeProto tensor_shape = 2; field
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | list_ops.cc | 168 shape_inference::ShapeHandle tensor_shape = c->UnknownShape(); in __anonebbd57d80502() local
317 shape_inference::ShapeHandle tensor_shape = c->input(0); in __anonebbd57d80902() local
351 shape_inference::ShapeHandle tensor_shape = c->input(0); in __anonebbd57d80a02() local
|
| D | data_flow_ops.cc | 701 ShapeHandle tensor_shape = shapes->at(0).shape; in __anon4a7ac08d1502() local
726 ShapeHandle tensor_shape = shapes->at(0).shape; in __anon4a7ac08d1602() local
766 ShapeHandle tensor_shape = shapes->at(0).shape; in __anon4a7ac08d1702() local
|
| /external/tensorflow/tensorflow/contrib/util/ |
| D | convert_graphdef_memmapped_format_lib.cc | 51 const TensorShapeProto tensor_shape = tensor_proto.tensor_shape(); in ConvertConstantsToImmutable() local
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | meta_graph.proto | 241 TensorShapeProto tensor_shape = 3; field
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | load_and_remap_matrix_op.cc | 138 TensorShape tensor_shape; in Compute() local
|
| D | parameterized_truncated_normal_op.cc | 350 TensorShape tensor_shape; in Compute() 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
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | linalg_ops.py | 231 def _use_composite_impl(fast, tensor_shape): argument
|
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_format.h | 455 inline int64 GetTensorDim(const TensorShape& tensor_shape, in GetTensorDim()
463 inline int64 GetFilterDim(const TensorShape& tensor_shape, in GetFilterDim()
|
| /external/tensorflow/tensorflow/lite/experimental/micro/ |
| D | simple_tensor_allocator_test.cc | 66 const int32_t tensor_shape[tensor_shape_size] = {size}; in Create1dTensor() local
|
| D | micro_interpreter_test.cc | 107 const int32_t tensor_shape[tensor_shape_size] = {1}; in BuildMockModel() local
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | costmodel.cc | 251 const TensorShapeProto& tensor_shape, in RecordMaxMemorySize()
554 Bytes CostModel::MinTensorMemoryUsage(const TensorShapeProto& tensor_shape, in MinTensorMemoryUsage()
|
| /external/tensorflow/tensorflow/contrib/periodic_resample/kernels/ |
| D | periodic_resample_op.h | 170 const tensorflow::TensorShape& tensor_shape) { in TensorShapeToVector()
|
| /external/tensorflow/tensorflow/contrib/mpi_collectives/kernels/ |
| D | mpi_ops.cc | 339 TensorShape tensor_shape = requests[0].tensor_shape(); in ConstructMPIResponse() local
362 TensorShape tensor_shape = requests[0].tensor_shape(); in ConstructMPIResponse() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | graph_utils.cc | 80 std::unique_ptr<tensorflow::TensorShapeProto> tensor_shape = in AddScalarConstNodeHelper() local
|