Home
last modified time | relevance | path

Searched defs:axis_size (Results 1 – 18 of 18) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/
H A DUnstackFixture.h68 const unsigned int axis_size = input_shape[axis_u]; in compute_target() local
96 const unsigned int axis_size = input_shape[axis_u]; in compute_reference() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/
H A Dstrided_slice_logic.h80 const int axis_size = input_shape.Dims(axis); in StartForAxis() local
129 const int axis_size = input_shape.Dims(axis); in StopForAxis() local
/aosp_15_r20/external/cronet/third_party/quic_trace/src/tools/render/
H A Daxis_renderer.cc79 size_t axis_size, in Ticks()
138 const vec2 axis_size = state_->window() - 2 * axis_offset; in Render() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dindex_ops.cc55 const int64_t axis_size = input_shape.dim_size(axis); in Compile() local
H A Droll_op.cc73 xla::XlaOp axis_size = xla::ConstantR0WithType( in Compile() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dgather.h50 const int axis_size = input_shape.Dims(axis); in Gather() local
H A Darg_min_max.h46 const int axis_size = input1_shape.Dims(axis); in ArgMinMax() local
/aosp_15_r20/external/freetype/src/sfnt/
H A Dsfobjs.c604 FT_UShort axis_size; in sfnt_init_face() local
/aosp_15_r20/external/libtextclassifier/native/tensorflow_models/seq_flow_lite/tflite_ops/
H A Dlayer_norm.cc144 const int* curr_pos, const int* axis, const int axis_size) { in GetOffset()
/aosp_15_r20/external/cronet/third_party/rust/chromium_crates_io/vendor/read-fonts-0.15.6/generated/
H A Dgenerated_fvar.rs100 pub fn axis_size(&self) -> u16 { in axis_size() method
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dgeneric_layout_optimizer_transposer.cc1470 const int axis_size = axis.size(); in IsAlongAxis() local
1734 int axis_size = axis.size(); in IsAlongAxis() local
/aosp_15_r20/external/freetype/src/type1/
H A Dt1load.c305 FT_Offset axis_size; in T1_Get_MM_Var() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/
H A Dquantization_utils.cc473 size_t axis_size = 1; in GetUniformQuantizedTypeForBias() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dreduce.cc87 TfLiteIntArray* axis_size = TfLiteIntArrayCreate(1); in ResizeTempAxis() local
/aosp_15_r20/external/ComputeLibrary/arm_compute/core/utils/misc/
H A DShapeCalculator.h1143 size_t axis_size = out_shape[axis]; in compute_split_shape() local
/aosp_15_r20/external/freetype/src/truetype/
H A Dttgxvar.c2326 FT_Offset axis_size; in TT_Get_MM_Var() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Doptimized_ops.h7969 int axis_size = input_shape.Dims(1); in ArgMinMaxLastAxis() local
7994 const int axis_size = input1_shape.Dims(axis); in ArgMinMax() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf.cc1676 auto axis_size = b.create<mhlo::ConstantOp>(b.getIntegerAttr( in matchAndRewrite() local