Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dtile.cc96 const int dimension_size, M multiplier, in CopyStringMultipleTimes()
117 const int dimension_size = in_dimensions.data[dimension]; in TileOneDimension() local
151 const int dimension_size = in_dimensions.data[dimension]; in TileStringOneDimension() local
H A Dpack.cc43 const int dimension_size = NumDimensions(input0) + 1; in Prepare() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Dresolve_constant_tile.cc43 const int dimension_size = in_dimensions.dims(dimension); in TileOneDimension() local
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DTile.cpp45 const int dimension_size = input_shape.dimensions[dimension]; in TileOneDimension() local
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/
H A Dragged_tensor_shape.py191 def dimension_size(self, axis): member in RaggedTensorDynamicShape
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A Ddynamic_shaped_ops.cc233 XlaOp operand, XlaOp dimension_size, in SetDimensionSizeWithRebound()
H A Darithmetic.cc148 int64_t dimension_size = input_shape.dimensions(axis); in ArgMinMax() local
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/distributions/
H A Dutil.py1202 def dimension_size(x, axis): function
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dcheck_ops.py1687 def _has_known_value(dimension_size): argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Ddynamic_padder.cc1976 auto dimension_size = in ConvertToDynamic() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner.cc1652 const int64_t dimension_size = in PatternMatchReshape() local
/aosp_15_r20/out/soong/.intermediates/external/federated-compute/federated-compute-cc-proto-lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/fcp/protos/
Dplan.pb.h10870 inline int SideChannel_SecureAggregand::dimension_size() const { in dimension_size() function