| /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 | mirror_pad_op.cc | 47 int64 dim_size = original_shape.dimensions(dimno); in DoMirrorPad() local
132 int64 dim_size = original_shape.dimensions(dimno); in DoMirrorPadGrad() local
|
| D | cross_op.cc | 52 for (auto dim_size : in0_shape.dim_sizes()) { in Compile() local
|
| D | spmd_manual_sharding_ops.cc | 55 int64 dim_size = in Compile() local
|
| D | strided_slice_op.cc | 148 xla::XlaOp dim_size; in EmitDynamicSlice() local
452 xla::XlaOp dim_size = xla::Slice(ctx->Input(0), {i}, {i + 1}, {1}); in CompileAsDynamicUpdateSlice() local
577 auto dim_size = xla::Slice(dynamic_shape, {dim}, {dim + 1}, {1}); in Compile() local
|
| D | while_op.cc | 527 int32 dim_size = shape.dimensions(0); in Compile() local
536 int32 dim_size = shape.dimensions(dim); in Compile() local
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_tensor_coding.cc | 101 int64 dim_size = val.shape().dim_size(d); in EncodeSkeleton() local
113 int64 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 dim_size = shape.dim_size(i); in Compute() local
83 int64 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 | 158 const auto dim_size = input.dim_size(d); in Compute() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_kernel.h | 226 bool is_dynamic_dimension(int64 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 | 197 const int dim_size = imagined_original_shape.size(); in Run() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | reduction_layout_normalizer.cc | 74 int64 dim_size = operand_shape.dimensions(logical_dim); in HandleReduce() 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 | 158 int64 dim_size = dim.size(); in CompareSymbolicallyShapedTensorSizes() 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 | 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 | 247 int64 dim_size = 0; in MakeShapeFromString() local
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | lhlo_legalize_to_gpu.cc | 63 auto dim_size = shaped_type.getDimSize(0); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/lite/tools/optimize/sparsity/ |
| D | format_converter.cc | 177 int dim_size = expanded_shape[original_dim_idx]; in DenseToSparse() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ |
| D | quantization_utils.cc | 271 void ExtractMinMaxFromAttr(DenseFPElementsAttr values, int dim_size, in ExtractMinMaxFromAttr()
356 int dim_size = shape[quant_dim]; in GetUniformQuantizedPerAxisTypeForWeight() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | export_utils.cc | 114 for (auto dim_size : attr.getShape()) { in ConvertAttribute() local
|