Searched refs:axis_size (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | strided_slice_logic.h | 96 int axis_size = input_shape.Dims(axis); in StartForAxis() local
98 start += axis_size; in StartForAxis()
102 start = Clamp(start, 0, axis_size - 1); in StartForAxis()
145 const int axis_size = input_shape.Dims(axis); in StopForAxis() local
147 stop += axis_size; in StopForAxis()
155 stop = Clamp(stop, 0, axis_size); in StopForAxis()
158 stop = Clamp(stop, -1, axis_size - 1); in StopForAxis()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | index_ops.cc | 55 const int64 axis_size = input_shape.dim_size(axis); in Compile() local
57 ctx, axis_size > 0, in Compile()
|
| D | index_ops_cpu.cc | 63 const int64 axis_size = input_shape.dim_size(axis); in Compile() local
64 OP_REQUIRES(ctx, axis_size > 0, in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | arithmetic.cc | 164 const int64 axis_size = ShapeUtil::GetDimension(input_shape, axis); in ArgMinMax() local
165 XlaOp iota = Iota(builder, output_type, axis_size); in ArgMinMax()
|
| /external/freetype/src/sfnt/ |
| D | sfobjs.c | 957 FT_UShort axis_size; in sfnt_init_face() local
976 FT_READ_USHORT( axis_size ) || in sfnt_init_face()
983 axis_size = 0; in sfnt_init_face()
990 axis_size != 20 || in sfnt_init_face()
999 axis_size * num_axes + in sfnt_init_face()
1036 default_value_offset += axis_size; in sfnt_init_face()
1040 instance_offset = array_start + axis_size * num_axes + 4; in sfnt_init_face()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reduce.cc | 78 TfLiteIntArray* axis_size = TfLiteIntArrayCreate(1); in ResizeTempAxis() local
79 axis_size->data[0] = static_cast<int>(NumElements(op_context->axis)); in ResizeTempAxis()
80 return context->ResizeTensor(context, resolved_axis, axis_size); in ResizeTempAxis()
|
| /external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/ |
| D | ops.py | 359 axis_size = lambda axis: -1 if axis.size is None else axis.size function
360 shape = [axis_size(axis) for axis in original_values[:start]]
366 shape.append(axis_size(axis))
367 shape.extend(axis_size(axis) for axis in original_values[stop:])
|
| /external/freetype/src/truetype/ |
| D | ttgxvar.c | 1985 FT_Offset axis_size; in TT_Get_MM_Var() local
2102 axis_size = ALIGN_SIZE( num_axes * in TT_Get_MM_Var()
2115 axis_size + in TT_Get_MM_Var()
2142 (FT_Var_Named_Style*)( (char*)mmvar->axis + axis_size ); in TT_Get_MM_Var()
2390 (FT_Var_Named_Style*)( (char*)mmvar->axis+ axis_size ); in TT_Get_MM_Var()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 2947 const int axis_size = input_shape.Dims(axis); in Gather() local
2963 TFLITE_DCHECK_LT(coords_data[i], axis_size); in Gather()
2966 input_data + (outer * axis_size + coords_data[i]) * inner_size, in Gather()
3878 const int axis_size = input1_shape.Dims(axis); in ArgMinMax() local
3895 auto min_max_value = input1_data[outer * axis_size * inner_size + inner]; in ArgMinMax()
3897 for (int i = 1; i < axis_size; ++i) { in ArgMinMax()
3899 input1_data[(outer * axis_size + i) * inner_size + inner]; in ArgMinMax()
|