| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | strided_slice_logic.h | 80 const int axis_size = input_shape.Dims(axis); in StartForAxis() local
81 if (axis_size == 0) { in StartForAxis()
102 start += axis_size; in StartForAxis()
108 start = Clamp(start, 0, axis_size); in StartForAxis()
111 start = Clamp(start, -1, axis_size - 1); in StartForAxis()
129 const int axis_size = input_shape.Dims(axis); in StopForAxis() local
130 if (axis_size == 0) { in StopForAxis()
160 stop += axis_size; in StopForAxis()
168 stop = Clamp(stop, 0, axis_size); in StopForAxis()
171 stop = Clamp(stop, -1, axis_size - 1); in StopForAxis()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | roll_op.cc | 73 xla::XlaOp axis_size = xla::ConstantR0WithType( in Compile() local
77 offset = ((offset % axis_size) + axis_size) % axis_size; in Compile()
85 start_indices[cur_axis] = axis_size - offset; in Compile()
|
| D | index_ops.cc | 57 const int64 axis_size = input_shape.dim_size(axis); in Compile() local
59 ctx, axis_size > 0, in Compile()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | arg_min_max.h | 35 const int axis_size = input1_shape.Dims(axis); in ArgMinMax() local
51 auto min_max_value = input1_data[outer * axis_size * inner_size + inner]; in ArgMinMax()
53 for (int i = 1; i < axis_size; ++i) { in ArgMinMax()
55 input1_data[(outer * axis_size + i) * inner_size + inner]; in ArgMinMax()
|
| D | gather.h | 37 const int axis_size = input_shape.Dims(axis); in Gather() local
53 TFLITE_DCHECK_LT(coords_data[i], axis_size); in Gather()
57 input_data + (outer * axis_size + coords_data[i]) * inner_size, in Gather()
|
| /external/llvm-project/mlir/lib/Dialect/Quant/Utils/ |
| D | FakeQuantSupport.cpp | 143 size_t axis_size = rmins.size(); in fakeQuantAttrsToType() local
144 if (axis_size != rmaxs.size()) { in fakeQuantAttrsToType()
146 << axis_size << " vs. " << rmaxs.size(), in fakeQuantAttrsToType()
162 scales.reserve(axis_size); in fakeQuantAttrsToType()
163 zeroPoints.reserve(axis_size); in fakeQuantAttrsToType()
164 for (size_t axis = 0; axis != axis_size; ++axis) { in fakeQuantAttrsToType()
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | keras_tensor.py | 459 for axis_size in ragged_spec.shape:
461 if axis_size is None or (
462 getattr(axis_size, 'value', True) is None):
465 prod = prod * axis_size
469 axis_size = ragged_spec.shape[axis]
470 if axis_size is None or (getattr(axis_size, 'value', True) is None):
479 rowlen = constant_op.constant(axis_size, ragged_spec.row_splits_dtype)
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | split_test.cc | 42 constexpr int axis_size = 1; in TestSplitTwoOutputsFloat() local
43 constexpr int tensors_size = input_size + output_size + axis_size; in TestSplitTwoOutputsFloat()
103 constexpr int axis_size = 1; in TestSplitFourOutputsFloat() local
104 constexpr int tensors_size = input_size + output_size + axis_size; in TestSplitFourOutputsFloat()
170 constexpr int axis_size = 1; in TestSplitTwoOutputsQuantized() local
171 constexpr int tensors_size = input_size + output_size + axis_size; in TestSplitTwoOutputsQuantized()
226 constexpr int axis_size = 1; in TestSplitTwoOutputsQuantized32() local
227 constexpr int tensors_size = input_size + output_size + axis_size; in TestSplitTwoOutputsQuantized32()
|
| D | split_v_test.cc | 53 constexpr int axis_size = 1; in TestSplitVFloat() local
58 input_size + output_size + axis_size + split_size; in TestSplitVFloat()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ |
| D | quantization_utils.cc | 381 size_t axis_size = 1; in GetUniformQuantizedTypeForBias() local
396 if ((axis_size != 1 && axis_size != type.getScales().size())) return {}; in GetUniformQuantizedTypeForBias()
399 axis_size = type.getScales().size(); in GetUniformQuantizedTypeForBias()
408 llvm::SmallVector<double, 4> scales(axis_size, 1.0); in GetUniformQuantizedTypeForBias()
415 for (int index = 0, e = axis_size; index != e; ++index) { in GetUniformQuantizedTypeForBias()
433 if (axis_size == 1) { in GetUniformQuantizedTypeForBias()
439 llvm::SmallVector<int64_t, 4> zero_points(axis_size, 0); in GetUniformQuantizedTypeForBias()
|
| /external/freetype/src/sfnt/ |
| D | sfobjs.c | 585 FT_UShort axis_size; in sfnt_init_face() local
604 FT_READ_USHORT( axis_size ) || in sfnt_init_face()
611 axis_size = 0; in sfnt_init_face()
618 axis_size != 20 || in sfnt_init_face()
627 axis_size * num_axes + in sfnt_init_face()
664 default_value_offset += axis_size; in sfnt_init_face()
668 instance_offset = array_start + axis_size * num_axes + 4; in sfnt_init_face()
|
| /external/libtextclassifier/native/tensorflow_models/seq_flow_lite/tflite_ops/ |
| D | layer_norm.cc | 144 const int* curr_pos, const int* axis, const int axis_size) { in GetOffset() argument
153 for (int redux = 0; redux < axis_size; ++redux) { in GetOffset()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reduce.cc | 84 TfLiteIntArray* axis_size = TfLiteIntArrayCreate(1); in ResizeTempAxis() local
85 axis_size->data[0] = static_cast<int>(NumElements(op_context->axis)); in ResizeTempAxis()
86 return context->ResizeTensor(context, resolved_axis, axis_size); in ResizeTempAxis()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer.cc | 1475 const int axis_size = axis.size(); in IsAlongAxis() local
1476 if (tensor.dims() != 1 || tensor.dim_size(0) != axis_size) { in IsAlongAxis()
1479 for (int i = 0; i < axis_size; ++i) { in IsAlongAxis()
1739 int axis_size = axis.size(); in IsAlongAxis() local
1742 } else if (list.i_size() != axis_size) { in IsAlongAxis()
1745 for (int i = 0; i < axis_size; ++i) { in IsAlongAxis()
|
| /external/freetype/src/truetype/ |
| D | ttgxvar.c | 2060 FT_Offset axis_size; in TT_Get_MM_Var() local
2177 axis_size = ALIGN_SIZE( num_axes * in TT_Get_MM_Var()
2190 axis_size + in TT_Get_MM_Var()
2217 (FT_Var_Named_Style*)( (char*)mmvar->axis + axis_size ); in TT_Get_MM_Var()
2465 (FT_Var_Named_Style*)( (char*)mmvar->axis+ axis_size ); in TT_Get_MM_Var()
|
| /external/freetype/src/type1/ |
| D | t1load.c | 316 FT_Offset axis_size; in T1_Get_MM_Var() local
335 axis_size = mmaster.num_axis * sizeof ( FT_Var_Axis ); in T1_Get_MM_Var()
339 axis_size ) ) in T1_Get_MM_Var()
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/ |
| D | np_array_ops.py | 921 axis_size = array_ops.shape(a, out_type=indices.dtype)[axis]
923 indices = clip_ops.clip_by_value(indices, 0, axis_size - 1)
925 indices = math_ops.floormod(indices, axis_size)
|