Home
last modified time | relevance | path

Searched refs:axis_ (Results 1 – 25 of 30) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Dquantize_and_dequantize_op.cc48 OP_REQUIRES_OK(ctx, ctx->GetAttr("axis", &axis_)); in QuantizeAndDequantizeV2Op()
75 ctx, (axis_ == -1 || axis_ < input.shape().dims()), in Compute()
76 errors::InvalidArgument("Shape must be at least rank ", axis_ + 1, in Compute()
78 const int depth = (axis_ == -1) ? 1 : input.dim_size(axis_); in Compute()
86 if (axis_ == -1) { in Compute()
97 " to match dim ", axis_, " of the input ", in Compute()
103 " to match dim ", axis_, " of the input ", in Compute()
107 auto range_shape = (axis_ == -1) ? TensorShape({}) : TensorShape({depth}); in Compute()
114 if (axis_ == -1) { in Compute()
122 input.template flat_inner_outer_dims<T, 3>(axis_ - 1), signed_input_, in Compute()
[all …]
Done_hot_op.cc48 OP_REQUIRES_OK(ctx, ctx->GetAttr("axis", &axis_)); in OneHotOp()
63 ctx, axis_ == -1 || (axis_ >= 0 && axis_ < output_dims), in Compute()
65 output_dims, "). But received: ", axis_)); in Compute()
76 const int axis = (axis_ == -1) ? indices_dims : axis_; in Compute()
124 int32 axis_; member in tensorflow::OneHotOp
Dquantize_op.cc104 OP_REQUIRES_OK(ctx, ctx->GetAttr("axis", &axis_)); in QuantizeV2Op()
115 if (axis_ > -1) { in Compute()
116 num_slices = input.dim_size(axis_); in Compute()
145 input.template flat_inner_outer_dims<float, 3>(axis_ - 1); in Compute()
147 for (int i = 0; i < axis_; ++i) { in Compute()
150 for (int i = axis_ + 1; i < output->dims(); ++i) { in Compute()
301 int axis_; member in tensorflow::QuantizeV2Op
Ddequantize_op.cc89 OP_REQUIRES_OK(ctx, ctx->GetAttr("axis", &axis_)); in DequantizeOp()
98 if (axis_ > -1) { in Compute()
99 num_slices = input.dim_size(axis_); in Compute()
116 for (int i = 0; i < axis_; ++i) { in Compute()
119 for (int i = axis_ + 1; i < float_output.dims(); ++i) { in Compute()
125 float_output.flat_inner_outer_dims<float, 3>(axis_ - 1); in Compute()
225 int axis_; member in tensorflow::DequantizeOp
Dpack_op.cc46 OP_REQUIRES_OK(context, context->GetAttr("axis", &axis_)); in PackOp()
54 int axis = axis_; in Compute()
58 errors::InvalidArgument("axis = ", axis_, " not in [", in Compute()
120 int axis_; member in tensorflow::PackOp
Dunpack_op.cc40 OP_REQUIRES_OK(context, context->GetAttr("axis", &axis_)); in UnpackOp()
48 int axis = axis_; in Compute()
52 errors::InvalidArgument("axis = ", axis_, " not in [", in Compute()
122 int axis_; member in tensorflow::UnpackOp
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Done_hot_op.cc29 OP_REQUIRES_OK(ctx, ctx->GetAttr("axis", &axis_)); in OneHotOp()
43 ctx, axis_ == -1 || (axis_ >= 0 && axis_ < output_dims), in Compile()
45 output_dims, "). But received: ", axis_)); in Compile()
56 const int axis = (axis_ == -1) ? indices_dims : axis_; in Compile()
74 int32 axis_; member in tensorflow::__anon2cdc51a70111::OneHotOp
Dpack_op.cc42 OP_REQUIRES_OK(ctx, ctx->GetAttr("axis", &axis_)); in PackOp()
64 int axis = axis_; in Compile()
68 errors::InvalidArgument("axis = ", axis_, " not in [", in Compile()
86 int axis_; member in tensorflow::__anon5a3e8bda0111::PackOp
Dunpack_op.cc42 OP_REQUIRES_OK(ctx, ctx->GetAttr("axis", &axis_)); in UnpackOp()
49 int axis = axis_; in Compile()
53 errors::InvalidArgument("axis = ", axis_, " not in [", in Compile()
85 int axis_; member in tensorflow::__anona049195d0111::UnpackOp
Dquantize_and_dequantize_op.cc53 OP_REQUIRES_OK(ctx, ctx->GetAttr("axis", &axis_)); in QuantizeAndDequantizeOp()
75 if (axis_ == -1) { in Compile()
86 ctx, axis_ >= 0 && axis_ < input_rank, in Compile()
90 if (i != axis_) { in Compile()
169 return xla::BroadcastInDim(op, input_dimensions, {axis_}); in Compile()
202 int axis_; member in tensorflow::__anon2fa8ad570111::QuantizeAndDequantizeOp
/external/tensorflow/tensorflow/python/kernel_tests/
Dnorm_op_test.py60 for axis_ in [], [1, 2, 3], [[1]], [[1], [2]], [3.1415], [1, 1]:
64 linalg_ops.norm(matrix, axis=axis_)
67 def _GetNormOpTest(dtype_, shape_, ord_, axis_, keep_dims_, use_static_shape_): argument
70 np_norm = np.linalg.norm(matrix, ord=ord_, axis=axis_, keepdims=keep_dims_)
75 tf_matrix, ord=ord_, axis=axis_, keepdims=keep_dims_)
80 tf_matrix, ord=ord_, axis=axis_, keepdims=keep_dims_)
86 is_matrix_norm = (isinstance(axis_, tuple) or
87 isinstance(axis_, list)) and len(axis_) == 2
92 if ord_ == "euclidean" or (axis_ is None and len(shape) > 2):
Dnormalize_op_test.py61 def _GetNormalizeOpTest(dtype_, shape_, ord_, axis_): argument
65 is_matrix_norm = (isinstance(axis_, tuple) or
66 isinstance(axis_, list)) and len(axis_) == 2
71 if ord_ == "euclidean" or (axis_ is None and len(shape) > 2):
76 tf_np_n, _ = self.evaluate(nn_impl.normalize(matrix, ord_, axis_))
77 np_n = _Normalize(matrix, ord_, axis_)
/external/rust/crates/criterion-plot/src/
Daxis.rs160 let axis_ = axis.display(); in script() localVariable
163 return format!("unset {}tics\n", axis_); in script()
165 format!("set {}tics nomirror ", axis_) in script()
175 script.push_str(&format!("set {}label '{}'\n", axis_, label)) in script()
179 script.push_str(&format!("set {}range [{}:{}]\n", axis_, low, high)) in script()
183 script.push_str(&format!("set logscale {}\n", axis_)); in script()
/external/tensorflow/tensorflow/lite/kernels/
Dsplit_v_test.cc40 axis_ = AddInput({TensorType_INT32, {1}}); in SplitVOpModel()
42 axis_ = AddConstInput(TensorType_INT32, {axis}, {1}); in SplitVOpModel()
51 {GetShape(input_), GetShape(size_splits_), GetShape(axis_)}); in SplitVOpModel()
64 void SetAxis(int axis) { PopulateTensor(axis_, {axis}); } in SetAxis()
75 int axis_; member in tflite::__anon3edbe8b00111::SplitVOpModel
Dcumsum_test.cc37 axis_ = AddInput({TensorType_INT32, {1}}); in CumsumOpModel()
44 BuildInterpreter({GetShape(input_), GetShape(axis_)}); in CumsumOpModel()
48 int axis() { return axis_; } in axis()
54 int axis_; member in tflite::ops::builtin::__anona08893e90111::CumsumOpModel
Dsplit_test.cc44 axis_ = AddInput({TensorType_INT32, {1}}); in SplitOpModel()
46 axis_ = AddConstInput(TensorType_INT32, {axis}, {1}); in SplitOpModel()
55 BuildInterpreter({GetShape(axis_), GetShape(input_)}); in SplitOpModel()
65 void SetAxis(int axis) { PopulateTensor(axis_, {axis}); } in SetAxis()
75 int axis_; member in tflite::__anon363ae4fb0111::SplitOpModel
Dexpand_dims_test.cc45 axis_ = AddInput(TensorType_INT32); in ExpandDimsOpModel()
49 axis_ = AddConstInput(TensorType_INT32, {axis}, {1}); in ExpandDimsOpModel()
59 PopulateTensor<int32_t>(axis_, {axis}); in ExpandDimsOpModel()
69 int axis_; member in tflite::__anon93b544bb0111::ExpandDimsOpModel
Darg_min_max_test.cc41 axis_ = in ArgBaseOpModel()
44 axis_ = AddConstInput(axis_type, {axis_value}, {1}); in ArgBaseOpModel()
47 axis_ = AddInput(axis_type); in ArgBaseOpModel()
53 int axis() const { return axis_; } in axis()
78 int axis_; member in tflite::__anon5be04bb10111::ArgBaseOpModel
Dreverse_test.cc34 axis_ = AddInput(axis); in ReverseOpModel()
44 int axis() { return axis_; } in axis()
51 int axis_; member in tflite::__anonef91558b0111::ReverseOpModel
Dreduce_test.cc33 void SetAxis(const std::vector<int>& data) { PopulateTensor(axis_, data); } in SetAxis()
75 int axis_; member in tflite::__anon88e4c2070111::BaseOpModel
86 axis_ = AddConstInput(TensorType_INT32, axis, axis_shape); in MeanOpConstModel()
100 axis_ = AddInput(axis); in MeanOpDynamicModel()
115 axis_ = AddConstInput(TensorType_INT32, axis, axis_shape); in SumOpConstModel()
129 axis_ = AddInput(axis); in SumOpDynamicModel()
144 axis_ = AddConstInput(TensorType_INT32, axis, axis_shape); in ProdOpConstModel()
158 axis_ = AddInput(axis); in ProdOpDynamicModel()
173 axis_ = AddConstInput(TensorType_INT32, axis, axis_shape); in MaxOpConstModel()
187 axis_ = AddInput(axis); in MaxOpDynamicModel()
[all …]
/external/tensorflow/tensorflow/python/ops/
Dsplit_benchmark.py113 axis_ = [1] # 0 is very fast because it doesn't actually do any copying
117 for axis in axis_:
Dconcat_benchmark.py136 axis_ = [0, 1]
141 for axis in axis_:
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Dsplit_test.cc25 axis_ = AddConstInput(TensorType_INT32, {axis}, {1}); in SplitOpModel()
50 int axis_; member in tflite::SplitOpModel
Dreduce_test.cc29 axis_ = AddConstInput(TensorType_INT32, axis, axis_shape); in ReduceOpModel()
48 int axis_; member in tflite::ReduceOpModel
/external/tensorflow/tensorflow/security/advisory/
Dtfsa-2020-027.md19 const int depth = (axis_ == -1) ? 1 : input.dim_size(axis_);

12