Home
last modified time | relevance | path

Searched refs:axis_value (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dexpand_dims.cc56 const TfLiteTensor& axis, int* axis_value) { in GetAxisValueFromTensor() argument
60 *axis_value = *GetTensorData<int32_t>(&axis); in GetAxisValueFromTensor()
63 *axis_value = *GetTensorData<int64_t>(&axis); in GetAxisValueFromTensor()
80 int axis_value; in Prepare() local
82 GetAxisValueFromTensor(context, *axis, &axis_value)); in Prepare()
83 return ExpandTensorDim(context, *input, axis_value, output); in Prepare()
95 int axis_value; in Eval() local
97 GetAxisValueFromTensor(context, *axis, &axis_value)); in Eval()
99 ExpandTensorDim(context, *input, axis_value, output)); in Eval()
Dsplit.cc51 int axis_value = GetTensorData<int>(axis)[0]; in ResizeOutputTensors() local
52 if (axis_value < 0) { in ResizeOutputTensors()
53 axis_value += NumDimensions(input); in ResizeOutputTensors()
56 const int input_size = SizeOfDimension(input, axis_value); in ResizeOutputTensors()
63 output_dims->data[axis_value] = slice_size; in ResizeOutputTensors()
109 int axis_value = GetTensorData<int>(op_context.axis)[0]; in Eval() local
110 if (axis_value < 0) { in Eval()
111 axis_value += NumDimensions(op_context.input); in Eval()
122 op_params.axis = axis_value; \ in Eval()
123 if (axis_value == 0) { \ in Eval()
Dsplit_v.cc62 int axis_value = GetTensorData<int>(axis)[0]; in ResizeOutputTensors() local
63 if (axis_value < 0) { in ResizeOutputTensors()
64 axis_value += NumDimensions(input); in ResizeOutputTensors()
93 const int input_size = SizeOfDimension(input, axis_value); in ResizeOutputTensors()
111 output_dims->data[axis_value] = size_splits_vector.at(i); in ResizeOutputTensors()
161 int axis_value = GetTensorData<int>(op_context.axis)[0]; in Eval() local
168 op_params.axis = axis_value; \ in Eval()
Darg_min_max.cc34 int axis_value = *GetTensorData<int>(axis); in ResizeOutput() local
35 if (axis_value < 0) { in ResizeOutput()
36 axis_value += NumDimensions(input); in ResizeOutput()
43 if (i != axis_value) { in ResizeOutput()
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dpfor.py1770 axis_value = tensor_util.constant_value(axis)
1771 if axis_value is not None:
1772 axis = axis_value
1815 axis_value = tensor_util.constant_value(axis)
1818 axis_value = int(axis)
1820 axis_value = None
1823 if axis_value is not None and axis_value != 0:
/external/tensorflow/tensorflow/core/ops/
Darray_ops.cc939 std::vector<int64> axis_value; in __anondb9326b21402() local
941 axis_value = AsInt64<int32>(axis_tensor, axis_tensor->NumElements()); in __anondb9326b21402()
943 axis_value = AsInt64<int64>(axis_tensor, axis_tensor->NumElements()); in __anondb9326b21402()
946 for (int i = 0; i < axis_value.size(); i++) { in __anondb9326b21402()
948 axis_value[i] < 0 ? rank + axis_value[i] : axis_value[i]; in __anondb9326b21402()
950 return errors::InvalidArgument("'axis'[", i, "] = ", axis_value[i], in __anondb9326b21402()
/external/tensorflow/tensorflow/lite/testing/
Dgenerate_examples.py3802 axis_value = tf.placeholder(dtype=tf.int32, name="axis", shape=[1])
3803 out = tf.expand_dims(input_value, axis=axis_value)
3804 return [input_value, axis_value], [out]
3809 axis_value = np.array([parameters["axis_value"]], dtype=np.int32)
3810 return [input_value, axis_value], sess.run(
3811 outputs, feed_dict=dict(zip(inputs, [input_value, axis_value])))