Home
last modified time | relevance | path

Searched defs:dimension_size (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dpack.cc56 const int dimension_size = NumDimensions(input0) + 1; in Prepare() local
Dtile.cc88 const int dimension_size = in_dimensions.data[dimension]; in TileOneDimension() local
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_constant_tile.cc43 const int dimension_size = in_dimensions.dims(dimension); in TileOneDimension() local
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_tensor_shape.py174 def dimension_size(self, axis): member in RaggedTensorDynamicShape
/external/tensorflow/tensorflow/python/ops/distributions/
Dutil.py1198 def dimension_size(x, axis): function