Searched defs:axis_shape (Results 1 – 7 of 7) sorted by relevance
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | ReverseFixture.h | 49 void setup(TensorShape shape, TensorShape axis_shape, DataType data_type) in setup()
70 …TensorType compute_target(const TensorShape &shape, const TensorShape &axis_shape, DataType data_t… in compute_target()
108 …SimpleTensor<T> compute_reference(const TensorShape &shape, const TensorShape &axis_shape, DataTyp… in compute_reference()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | roll_op.cc | 32 const TensorShape axis_shape = ctx->InputShape(2); in Compile() local
|
| D | quantize_and_dequantize_op.cc | 162 xla::Shape axis_shape = b->GetShape(min_range).ValueOrDie(); in Compile() local
|
| D | gather_op.cc | 164 const TensorShape axis_shape = context->InputShape(2); in XlaGatherWithBatchDimsOpImpl() local
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | reduce_test.cc | 26 const TensorData& output, std::initializer_list<int> axis_shape, in ReduceOpModel()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reduce_test.cc | 84 std::initializer_list<int> axis_shape, in BaseConstOpModel()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 4714 std::initializer_list<int> axis_shape, in MeanOpConstModel()
|