Searched defs:axis_shape (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
| H A 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()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | roll_op.cc | 32 const TensorShape axis_shape = ctx->InputShape(2); in Compile() local
|
| H A D | quantize_and_dequantize_op.cc | 162 xla::Shape axis_shape = b->GetShape(min_range).ValueOrDie(); in Compile() local
|
| H A D | gather_op.cc | 165 const TensorShape axis_shape = context->InputShape(2); in XlaGatherWithBatchDimsOpImpl() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | reduce_test_common.h | 75 std::initializer_list<int> axis_shape, in BaseConstOpModel()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| H A D | reduce_test.cc | 28 const TensorData& output, std::initializer_list<int> axis_shape, in ReduceOpModel()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| H A D | nnapi_delegate_test.cc | 4791 std::initializer_list<int> axis_shape, in MeanOpConstModel()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | randomized_tests.cc | 3994 auto axis_shape = RandomDims(1, 1, 1, rank + 1); in TEST_F() local
|