| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | shape_util.cc | 30 int64_t dim_size = input_shape.dim_size(i); in TensorShapeToConstant() local
41 int64_t dim_size = input_shape.dim_size(i); in TensorShapeToConstant() local
|
| D | mirror_pad_op.cc | 47 int64_t dim_size = original_shape.dimensions(dimno); in DoMirrorPad() local
133 int64_t dim_size = original_shape.dimensions(dimno); in DoMirrorPadGrad() local
|
| D | cross_op.cc | 54 for (auto dim_size : in0_shape.dim_sizes()) { in Compile() local
|
| D | strided_slice_op.cc | 155 xla::XlaOp dim_size; in EmitDynamicSlice() local
489 xla::XlaOp dim_size = xla::Slice(ctx->Input(0), {i}, {i + 1}, {1}); in CompileAsDynamicUpdateSlice() local
616 auto dim_size = xla::Slice(dynamic_shape, {dim}, {dim + 1}, {1}); in Compile() local
|
| D | sharding_util_ops.cc | 209 const int64 dim_size = input_shape.dim_size(dim); in CompileInternal() local
350 const int64_t dim_size = output_shape[dim]; in CompileInternal() local
|
| D | while_op.cc | 540 int32_t dim_size = shape.dimensions(0); in Compile() local
549 int32_t dim_size = shape.dimensions(dim); in Compile() local
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_tensor_coding.cc | 98 int64_t dim_size = val.shape().dim_size(d); in EncodeSkeleton() local
110 int64_t dim_size = val.shape().dim_size(d); in EncodeSkeleton() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | roll_op.cc | 81 gtl::InlinedVector<int32, 4> dim_size(num_dims); in Compute() local
122 const int num_dims, const gtl::ArraySlice<int32> dim_size, in DoRoll()
186 const gtl::ArraySlice<int32> dim_size, const T* input, in DoRollWithMemcpy()
|
| D | shape_ops.h | 54 int64_t dim_size = shape.dim_size(i); in Compute() local
83 int64_t dim_size = shape.dim_size(j); in Compute() local
|
| D | roll_op_gpu.cu.cc | 35 const int32* __restrict__ dim_size, in RollKernel()
|
| D | transpose_op.cc | 157 const auto dim_size = input.dim_size(d); in Compute() local
|
| /external/ComputeLibrary/src/core/utils/helpers/ |
| D | tensor_transform.cpp | 60 const int dim_size = input_shape[index]; in calculate_start_on_index() local
108 const int dim_size = input_shape[index]; in calculate_end_on_index() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_utils.h | 65 int64_t dim_size = xla_ty.dimensions(dim); in ConvertTensorShapeToType() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_kernel.h | 276 bool is_dynamic_dimension(int64_t dim_size) { return dim_size == -1; } in is_dynamic_dimension()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_constant_slice.cc | 54 int dim_size = size[i]; in Slice() local
|
| D | identify_nearest_upsample.cc | 198 const int dim_size = imagined_original_shape.size(); in Run() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | shape_partition.cc | 117 const int64_t 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
|
| /external/tensorflow/tensorflow/cc/experimental/base/public/ |
| D | tensor.h | 131 inline int64_t Tensor::dim_size(int d) const { in dim_size() function
|
| /external/tensorflow/tensorflow/python/lib/core/ |
| D | ndarray_tensor_bridge.cc | 195 Status ArrayFromMemory(int dim_size, npy_intp* dims, void* data, DataType dtype, in ArrayFromMemory()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | reduction_layout_normalizer.cc | 101 int64_t dim_size = operand_shape.dimensions(logical_dim); in HandleReduce() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference_testutil.cc | 247 int64_t dim_size = 0; in MakeShapeFromString() local
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_embedding_configuration_proto_rewrite.cc | 94 for (const int dim_size : input_shape) { in ValidateFeatureDescriptors() local
|
| /external/tensorflow/tensorflow/lite/kernels/shim/ |
| D | test_util.cc | 93 const size_t dim_size = tensor->dims->data[axis]; in TfliteTensorDebugStringImpl() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | dynamic_shaped_ops.cc | 266 auto dim_size = xla::Slice(size_vector, {i}, {i + 1}, {1}); in SetAllDimensionSizes() local
|