Searched refs:shape_dim (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_sharding.cc | 278 const int64 shape_dim = shape.dimensions(i); in TileOffsetForDevice() local
280 index[i] * CeilOfRatio(shape_dim, tile_assignment_.dim(i)), shape_dim); in TileOffsetForDevice()
299 const int64 shape_dim = shape.dimensions(i); in TileLimitForDevice() local
301 (index[i] + 1) * CeilOfRatio(shape_dim, tile_assignment_.dim(i)), in TileLimitForDevice()
302 shape_dim); in TileLimitForDevice()
587 const int64 shape_dim = shape.dimensions(i); in TileShape() local
589 index[i] * CeilOfRatio(shape_dim, tile_assignment_.dim(i)), shape_dim); in TileShape()
591 (index[i] + 1) * CeilOfRatio(shape_dim, tile_assignment_.dim(i)), in TileShape()
592 shape_dim); in TileShape()
|
| D | dynamic_padder.cc | 764 int64 shape_dim = window_dim_to_shape_dim(dim_index); in RewriteInputWithDynamicPadding() local
767 auto* padding_dim = padding_configs.mutable_dimensions(shape_dim); in RewriteInputWithDynamicPadding()
785 start_indices[shape_dim] = slicing_start; in RewriteInputWithDynamicPadding()
787 padded_shape.mutable_dimensions()[shape_dim] = in RewriteInputWithDynamicPadding()
789 window_util::DilatedBound(padded_shape.dimensions(shape_dim), in RewriteInputWithDynamicPadding()
795 input->mutable_shape()->set_dynamic_dimension(shape_dim, false); in RewriteInputWithDynamicPadding()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | import_tensorflow_test.cc | 92 auto shape_dim = shape->add_dim(); in BuildNode() local
93 shape_dim->set_size(output_shape_dim); in BuildNode()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference.cc | 777 DimensionHandle shape_dim = Dim(tensor_shape, 0); in InternalMakeShapeFromTensor() local
778 if (!ValueKnown(shape_dim)) { in InternalMakeShapeFromTensor()
781 const auto num_dims = Value(shape_dim); in InternalMakeShapeFromTensor()
|
| /external/tensorflow/tensorflow/python/debug/lib/ |
| D | debug_data.py | 124 for shape_dim in shape:
125 num_elements *= shape_dim
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 1964 @RegisterPForWithArgs("Conv2DBackpropInput", flatten_dims=[2], shape_dim=0)
1965 @RegisterPForWithArgs("AvgPoolGrad", flatten_dims=[1], shape_dim=0)
1966 @RegisterPForWithArgs("AvgPool3DGrad", flatten_dims=[1], shape_dim=0)
1968 shape_dim): argument
1973 ones = array_ops.ones([array_ops.shape(inputs[shape_dim])[0] - 1],
1975 inputs[shape_dim] *= array_ops.concat([n, ones], axis=0)
|