Home
last modified time | relevance | path

Searched defs:shape_dim (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding.cc174 const int64 shape_dim = shape.dimensions(i); in TileOffsetForDevice() local
193 const int64 shape_dim = shape.dimensions(i); in TileLimitForDevice() local
/external/tensorflow/tensorflow/lite/toco/
Dimport_tensorflow_test.cc90 auto shape_dim = shape->add_dim(); in BuildNode() local
/external/tensorflow/tensorflow/core/framework/
Dshape_inference.cc833 DimensionHandle shape_dim = Dim(tensor_shape, 0); in InternalMakeShapeFromTensor() local
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dpfor.py1569 shape_dim): argument