Searched defs:shape_output (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
| H A D | GEMMLowpFixture.h | 235 …void setup(TensorShape shape_a, TensorShape shape_b, TensorShape shape_output, int32_t a_offset, i… in setup()
242 …rget(const TensorShape &shape_a, const TensorShape &shape_b, const TensorShape &shape_output, int3… in compute_target()
248 …ence(const TensorShape &shape_a, const TensorShape &shape_b, const TensorShape &shape_output, int3… in compute_reference()
262 …void setup(TensorShape shape_a, TensorShape shape_b, TensorShape shape_output, int32_t a_offset, i… in setup()
301 …ence(const TensorShape &shape_a, const TensorShape &shape_b, const TensorShape &shape_output, int3… in compute_reference()
336 …void setup(TensorShape shape_a, TensorShape shape_b, TensorShape shape_output, int32_t a_offset, i… in setup()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | eval_const_tensor.cc | 176 Tensor shape_output; in TryToInferTensorOutputFromStridedSliceNode() local
|