Searched refs:axis_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reduce_test.cc | 61 std::initializer_list<int> axis_shape, in MeanOpConstModel() argument
64 axis_ = AddConstInput(TensorType_INT32, axis, axis_shape); in MeanOpConstModel()
90 std::initializer_list<int> axis_shape, in SumOpConstModel() argument
93 axis_ = AddConstInput(TensorType_INT32, axis, axis_shape); in SumOpConstModel()
119 std::initializer_list<int> axis_shape, in ProdOpConstModel() argument
122 axis_ = AddConstInput(TensorType_INT32, axis, axis_shape); in ProdOpConstModel()
148 std::initializer_list<int> axis_shape, in MaxOpConstModel() argument
151 axis_ = AddConstInput(TensorType_INT32, axis, axis_shape); in MaxOpConstModel()
177 std::initializer_list<int> axis_shape, in MinOpConstModel() argument
180 axis_ = AddConstInput(TensorType_INT32, axis, axis_shape); in MinOpConstModel()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | gather_op.cc | 161 const TensorShape axis_shape = context->InputShape(2); in Compile() local
162 OP_REQUIRES(context, TensorShapeUtils::IsScalar(axis_shape), in Compile()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 3422 std::initializer_list<int> axis_shape, in MeanOpConstModel() argument
3425 axis_ = AddConstInput(TensorType_INT32, axis, axis_shape); in MeanOpConstModel()
|