Home
last modified time | relevance | path

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

/external/ComputeLibrary/tests/validation/fixtures/
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
/external/tensorflow/tensorflow/lite/kernels/internal/
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
/external/cronet/third_party/quic_trace/src/tools/render/
Daxis_renderer.cc79 size_t axis_size, in Ticks()
138 const vec2 axis_size = state_->window() - 2 * axis_offset; in Render() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Droll_op.cc73 xla::XlaOp axis_size = xla::ConstantR0WithType( in Compile() local
Dindex_ops.cc55 const int64_t axis_size = input_shape.dim_size(axis); in Compile() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dgather.h50 const int axis_size = input_shape.Dims(axis); in Gather() local
Darg_min_max.h46 const int axis_size = input1_shape.Dims(axis); in ArgMinMax() local
/external/freetype/src/sfnt/
Dsfobjs.c604 FT_UShort axis_size; in sfnt_init_face() local
/external/libtextclassifier/native/tensorflow_models/seq_flow_lite/tflite_ops/
Dlayer_norm.cc144 const int* curr_pos, const int* axis, const int axis_size) { in GetOffset()
/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/
Dquantization_utils.cc473 size_t axis_size = 1; in GetUniformQuantizedTypeForBias() local
/external/freetype/src/type1/
Dt1load.c301 FT_Offset axis_size; in T1_Get_MM_Var() local
/external/tensorflow/tensorflow/lite/kernels/
Dreduce.cc87 TfLiteIntArray* axis_size = TfLiteIntArrayCreate(1); in ResizeTempAxis() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_transposer.cc1470 const int axis_size = axis.size(); in IsAlongAxis() local
1734 int axis_size = axis.size(); in IsAlongAxis() local
/external/ComputeLibrary/arm_compute/core/utils/misc/
DShapeCalculator.h1143 size_t axis_size = out_shape[axis]; in compute_split_shape() local
/external/freetype/src/truetype/
Dttgxvar.c2291 FT_Offset axis_size; in TT_Get_MM_Var() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
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
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc1676 auto axis_size = b.create<mhlo::ConstantOp>(b.getIntegerAttr( in matchAndRewrite() local