| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | one_hot_op.cc | 29 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::__anonc3bfa5090111::OneHotOp
|
| D | pack_op.cc | 42 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::__anon22a9ca7c0111::PackOp
|
| D | unpack_op.cc | 42 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::__anon307e887f0111::UnpackOp
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | norm_op_test.py | 60 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):
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | one_hot_op.cc | 47 OP_REQUIRES_OK(ctx, ctx->GetAttr("axis", &axis_)); in OneHotOp()
62 ctx, axis_ == -1 || (axis_ >= 0 && axis_ < output_dims), in Compute()
64 output_dims, "). But received: ", axis_)); in Compute()
75 const int axis = (axis_ == -1) ? indices_dims : axis_; in Compute()
123 int32 axis_; member in tensorflow::OneHotOp
|
| D | unpack_op.cc | 43 OP_REQUIRES_OK(context, context->GetAttr("axis", &axis_)); in UnpackOp()
51 int axis = axis_; in Compute()
55 errors::InvalidArgument("axis = ", axis_, " not in [", in Compute()
126 int axis_; member in tensorflow::UnpackOp
|
| D | pack_op.cc | 49 OP_REQUIRES_OK(context, context->GetAttr("axis", &axis_)); in PackOp()
67 int axis = axis_; in Compute()
71 errors::InvalidArgument("axis = ", axis_, " not in [", in Compute()
132 int axis_; member in tensorflow::PackOp
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | split_test.cc | 33 axis_ = AddInput({TensorType_INT32, {1}}); in SplitOpModel()
35 axis_ = AddConstInput(TensorType_INT32, {axis}, {1}); in SplitOpModel()
44 BuildInterpreter({GetShape(axis_), GetShape(input_)}); in SplitOpModel()
54 void SetAxis(int axis) { PopulateTensor(axis_, {axis}); } in SetAxis()
64 int axis_; member in tflite::__anona5eaa19d0111::SplitOpModel
|
| D | split_v_test.cc | 36 axis_ = AddInput({TensorType_INT32, {1}}); in SplitVOpModel()
38 axis_ = AddConstInput(TensorType_INT32, {axis}, {1}); in SplitVOpModel()
47 {GetShape(input_), GetShape(size_splits_), GetShape(axis_)}); in SplitVOpModel()
60 void SetAxis(int axis) { PopulateTensor(axis_, {axis}); } in SetAxis()
71 int axis_; member in tflite::__anon596d55d20111::SplitVOpModel
|
| D | expand_dims_test.cc | 33 axis_ = AddInput(TensorType_INT32); in ExpandDimsOpModel()
42 void SetAxis(int axis) { PopulateTensor<int32_t>(axis_, {axis}); } in SetAxis()
48 int axis_; member in tflite::__anon459c72dd0111::ExpandDimsOpModel
|
| D | reverse_test.cc | 31 axis_ = AddInput(axis); in ReverseOpModel()
41 int axis() { return axis_; } in axis()
48 int axis_; member in tflite::__anon0a22c2ad0111::ReverseOpModel
|
| D | reduce_test.cc | 29 void SetAxis(const std::vector<int>& data) { PopulateTensor(axis_, data); } in SetAxis()
53 int axis_; member in tflite::__anonee8c12e90111::BaseOpModel
64 axis_ = AddConstInput(TensorType_INT32, axis, axis_shape); in MeanOpConstModel()
78 axis_ = AddInput(axis); in MeanOpDynamicModel()
93 axis_ = AddConstInput(TensorType_INT32, axis, axis_shape); in SumOpConstModel()
107 axis_ = AddInput(axis); in SumOpDynamicModel()
122 axis_ = AddConstInput(TensorType_INT32, axis, axis_shape); in ProdOpConstModel()
136 axis_ = AddInput(axis); in ProdOpDynamicModel()
151 axis_ = AddConstInput(TensorType_INT32, axis, axis_shape); in MaxOpConstModel()
165 axis_ = AddInput(axis); in MaxOpDynamicModel()
[all …]
|
| D | arg_min_max_test.cc | 32 axis_ = AddInput(TensorType_INT32); in ArgBaseOpModel()
37 int axis() { return axis_; } in axis()
44 int axis_; member in tflite::__anon0dc779d30111::ArgBaseOpModel
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | split_benchmark.py | 113 axis_ = [1] # 0 is very fast because it doesn't actually do any copying
117 for axis in axis_:
|
| D | concat_benchmark.py | 136 axis_ = [0, 1]
141 for axis in axis_:
|
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/ |
| D | vector_diffeomixture.py | 1059 axis_ = tensor_util.constant_value(axis)
1060 if axis_ is not None:
1061 axis = np.int(ndims + axis_ if axis_ < 0 else axis_)
|
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | util.py | 1312 axis_ = tensor_util.constant_value(axis)
1313 if axis_ is not None:
1314 axis = axis_
1318 if axis_ >= 0 or x.shape.ndims is not None:
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 3391 void SetAxis(const std::vector<int>& data) { PopulateTensor(axis_, data); } in SetAxis()
3414 int axis_; member in tflite::__anon6d004b400111::BaseReduceOpModel
3425 axis_ = AddConstInput(TensorType_INT32, axis, axis_shape); in MeanOpConstModel()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 2514 squeeze_attrs.axis_ = gtl::ArraySlice<int>(axis); // non-absl ok in TEST_F()
|