Home
last modified time | relevance | path

Searched refs:dimension_size (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_tensor_shape.py112 for axis, dimension_size in enumerate(partitioned_dim_sizes):
113 if dimension_size.shape.ndims is None:
116 dimension_size.shape.with_rank_at_most(1)
174 def dimension_size(self, axis): member in RaggedTensorDynamicShape
309 math_ops.equal(lengths, self.dimension_size(axis)))
311 axis_dim_size = self.dimension_size(axis)
321 self.dimension_size(axis)
354 return self.dimension_size(axis) * self.num_slices_in_dimension(axis - 1)
358 axis_dim_size = self.dimension_size(axis)
387 dim_size = self.dimension_size(axis)
[all …]
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_constant_tile.cc43 const int dimension_size = in_dimensions.dims(dimension); in TileOneDimension() local
45 CopyMultipleTimes(in_data, dimension_size, multipliers[dimension], in TileOneDimension()
48 dimension_size, in TileOneDimension()
49 dimension_size * static_cast<int>(multipliers[dimension])); in TileOneDimension()
54 for (int i = 0; i < dimension_size; ++i) { in TileOneDimension()
/external/tensorflow/tensorflow/lite/kernels/
Dtile.cc88 const int dimension_size = in_dimensions.data[dimension]; in TileOneDimension() local
90 CopyMultipleTimes(in_data, dimension_size, multipliers[dimension], in TileOneDimension()
93 dimension_size, in TileOneDimension()
94 dimension_size * static_cast<int>(multipliers[dimension])); in TileOneDimension()
99 for (int i = 0; i < dimension_size; ++i) { in TileOneDimension()
Dpack.cc56 const int dimension_size = NumDimensions(input0) + 1; in Prepare() local
58 TfLiteIntArray* output_shape = TfLiteIntArrayCreate(dimension_size); in Prepare()
60 for (int index = 0; index < dimension_size; ++index) { in Prepare()
/external/tensorflow/tensorflow/python/ops/
Dstate_ops.py897 dimension_size = indices_shape[dimension]
899 shape_to_broadcast[dimension] = dimension_size
901 gen_math_ops._range(0, dimension_size, 1), shape_to_broadcast)
/external/tensorflow/tensorflow/contrib/distributions/python/ops/
Dmvn_tril.py203 num_rows=distribution_util.dimension_size(loc, -1),
/external/tensorflow/tensorflow/contrib/distributions/python/ops/bijectors/
Daffine.py304 shape_hint = distribution_util.dimension_size(perturb_factor, axis=-2)
/external/tensorflow/tensorflow/python/ops/distributions/
Dutil.py1198 def dimension_size(x, axis): function
1264 dimension_size(probs, axis=-1),
1265 dimension_size(grid, axis=-1),
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Doperation_semantics.md1022 start_indices[i] = clamp(start_indices[i], 0, operand.dimension_size[i] - size_indices[i])
1088 start_indices[i] = clamp(start_indices[i], 0, operand.dimension_size[i] - update.dimension_size[i])