| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | shape_util.cc | 30 int64 dim_size = input_shape.dim_size(i); in TensorShapeToConstant() local
41 int64 dim_size = input_shape.dim_size(i); in TensorShapeToConstant() local
|
| D | cross_op.cc | 52 for (auto dim_size : in0_shape.dim_sizes()) { in Compile() local
|
| D | mirror_pad_op.cc | 45 int64 dim_size = original_shape.dimensions(dimno); in DoMirrorPad() local
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_tensor_coding.cc | 99 int64 dim_size = val.shape().dim_size(d); in EncodeSkeleton() local
111 int64 dim_size = val.shape().dim_size(d); in EncodeSkeleton() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | roll_op.cc | 40 const int num_dims, const gtl::ArraySlice<int>& dim_size, in DoRoll()
103 const int num_dims, const gtl::ArraySlice<int>& dim_size, in DoRollWithMemcpy()
270 gtl::InlinedVector<int, 4> dim_size(num_dims); in Compute() local
|
| D | shape_ops.h | 52 int64 dim_size = shape.dim_size(i); in Compute() local
81 int64 dim_size = shape.dim_size(j); in Compute() local
|
| D | transpose_op.cc | 172 const auto dim_size = input.dim_size(d); in Compute() local
|
| D | set_kernels.cc | 105 const auto dim_size = sparse_tensor_shape[j]; in CheckGroup() local
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_constant_slice.cc | 54 int dim_size = size[i]; in Slice() local
|
| D | propagate_fixed_sizes.cc | 1686 int dim_size = in ProcessStridedSliceOperator() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | shape_partition.cc | 116 const int64 dim_size = shape_.dimensions(dimensions_[i]); in ShapePartitionIterator() local
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | symbolic_shapes_test.cc | 34 for (int dim_size : dims) { in MakeShape() local
|
| D | symbolic_shapes.cc | 155 int64 dim_size = dim.size(); in CompareSymbolicallyShapedTensorSizes() local
|
| /external/tensorflow/tensorflow/python/lib/core/ |
| D | ndarray_tensor_bridge.cc | 189 Status ArrayFromMemory(int dim_size, npy_intp* dims, void* data, DataType dtype, in ArrayFromMemory()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference_testutil.cc | 249 int64 dim_size = 0; in MakeShapeFromString() local
|
| D | tensor_shape.cc | 272 int64 TensorShapeBase<Shape>::dim_size(int d) const { in dim_size() function in tensorflow::TensorShapeBase
|
| D | tensor.h | 210 int64 dim_size(int d) const { return shape().dim_size(d); } in dim_size() function
|
| D | tensor_shape_test.cc | 269 int64 dim_size(int d) const { in dim_size() function in tensorflow::__anon515713880111::TensorShapeOld
|
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_format.h | 515 auto dim_size = spatial[dim]; in ShapeFromFormat() local
|
| D | mkl_util.h | 243 int64 dim_size(int index) const { return sizes_[index]; } in dim_size() function
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tensor.cc | 1077 tensorflow::int64 dim_size = TFE_TensorDebugInfoOnDeviceDim(debug_info, i); in TFE_Py_TensorShapeOnDevice() local
|
| D | pywrap_tfe_src.cc | 2906 for (tensorflow::int64 dim_size : tensor_shape.dim_sizes()) { local
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | sparse_tensor.h | 662 int64 dim_size = start[dim] + size[dim] < output_shape.dim_size(dim) in Slice() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator_typed_visitor.h | 1339 const int64 dim_size = lhs->shape().dimensions(lhs_dnum); in HandleDotSlowPath() local
|