Home
last modified time | relevance | path

Searched defs:axis_value (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dexpand_dims.cc56 const TfLiteTensor& axis, int* axis_value) { in GetAxisValueFromTensor()
83 int axis_value; in Prepare() local
101 int axis_value; in Eval() local
Dsplit.cc54 int axis_value = GetTensorData<int>(axis)[0]; in ResizeOutputTensors() local
118 int axis_value = GetTensorData<int>(op_context.axis)[0]; in Eval() local
Darg_min_max_test.cc33 ArgBaseOpModel(TensorType input_type, int axis_value, TensorType axis_type, in ArgBaseOpModel()
85 int axis_value, TensorType axis_type, bool constant_axis, in ArgMaxOpModel()
100 int axis_value, TensorType axis_type, bool constant_axis, in ArgMinOpModel()
Dsplit_v.cc68 int axis_value = GetTensorData<int>(axis)[0]; in ResizeOutputTensors() local
171 int axis_value = GetTensorData<int>(op_context.axis)[0]; in Eval() local
Darg_min_max.cc40 int axis_value; in ResizeOutput() local
/external/tensorflow/tensorflow/lite/micro/kernels/
Dsplit_v.cc30 const TfLiteEvalTensor* input, int axis_value) { in SplitImpl()
91 int axis_value = tflite::micro::GetTensorData<int32_t>(axis)[0]; in Eval() local
Dsplit.cc29 const TfLiteEvalTensor* input, int axis_value) { in SplitImpl()
86 int axis_value = tflite::micro::GetTensorData<int32_t>(axis)[0]; in Eval() local
Dsplit_v_test.cc334 int32_t axis_value[] = {0}; in TF_LITE_MICRO_TEST() local
411 int32_t axis_value[] = {0}; in TF_LITE_MICRO_TEST() local
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Darg_min_max_builder.cc48 int axis_value = axis.data.i32[0]; in PopulateSubGraph() local
Dsplit_builder.cc46 int axis_value = axis.data.i32[0] + (4 - input_tensor.dims->size); in PopulateSubGraph() local
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_device_selection_test.cc245 int axis_value, TensorType output_type, const NnApi* nnapi, in ArgMaxOpModel()
252 int axis_value, TensorType output_type, const NnApi* nnapi, in ArgMaxOpModel()
266 int axis_value, TensorType output_type) { in Init()
/external/harfbuzz_ng/src/
Dhb-ot-stat-table.hh268 const AxisValue &axis_value = (this + get_axis_value_offsets ()[axis_value_index]); in get_axis_value_name_id() local
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc1179 int32_t axis_value = op.axis(); in Verify() local
1792 int64_t axis_value = op.axis().getInt(); in inferReturnTypes() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlower_tf.cc747 auto axis_value = rewriter.create<ConstOp>( in matchAndRewrite() local
/external/tensorflow/tensorflow/core/ops/
Darray_ops.cc1006 std::vector<int64> axis_value; in __anon42d741191402() local