Searched refs:shape_dim (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/contrib/slim/python/slim/data/ |
| D | tfexample_decoder.py | 196 shape_dim = keys_to_tensors[k]
197 if isinstance(shape_dim, sparse_tensor.SparseTensor):
198 shape_dim = sparse_ops.sparse_tensor_to_dense(shape_dim)
199 shape_dims.append(shape_dim)
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_sharding.cc | 174 const int64 shape_dim = shape.dimensions(i); in TileOffsetForDevice() local
176 index[i] * CeilOfRatio(shape_dim, tile_assignment_.dim(i)), shape_dim); in TileOffsetForDevice()
193 const int64 shape_dim = shape.dimensions(i); in TileLimitForDevice() local
195 (index[i] + 1) * CeilOfRatio(shape_dim, tile_assignment_.dim(i)), in TileLimitForDevice()
196 shape_dim); in TileLimitForDevice()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | import_tensorflow_test.cc | 90 auto shape_dim = shape->add_dim(); in BuildNode() local
91 shape_dim->set_size(output_shape_dim); in BuildNode()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference.cc | 833 DimensionHandle shape_dim = Dim(tensor_shape, 0); in InternalMakeShapeFromTensor() local
834 if (!ValueKnown(shape_dim)) { in InternalMakeShapeFromTensor()
837 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 | 1566 @RegisterPForWithArgs("Conv2DBackpropInput", flatten_dims=[2], shape_dim=0)
1567 @RegisterPForWithArgs("AvgPoolGrad", flatten_dims=[1], shape_dim=0)
1569 shape_dim): argument
1575 [array_ops.shape(inputs[shape_dim])[0] - 1], dtype=n.dtype)
1576 inputs[shape_dim] *= array_ops.concat([n, ones], axis=0)
|