| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | GEMMInterleave4x4Fixture.h | 83 …TensorType compute_target(const TensorShape &shape_a, const TensorShape &shape_b, DataType data_ty… in compute_target()
113 …SimpleTensor<T> compute_reference(const TensorShape &shape_a, const TensorShape &shape_b, DataType… in compute_reference()
|
| D | GEMMTranspose1xWFixture.h | 84 …TensorType compute_target(const TensorShape &shape_a, const TensorShape &shape_b, DataType data_ty… in compute_target()
|
| D | GEMMLowpFixture.h | 235 …void setup(TensorShape shape_a, TensorShape shape_b, TensorShape shape_output, int32_t a_offset, i… in setup()
242 …TensorType compute_target(const TensorShape &shape_a, const TensorShape &shape_b, const TensorShap… in compute_target()
248 …SimpleTensor<int32_t> compute_reference(const TensorShape &shape_a, const TensorShape &shape_b, co… in compute_reference()
262 …void setup(TensorShape shape_a, TensorShape shape_b, TensorShape shape_output, int32_t a_offset, i… in setup()
301 …SimpleTensor<TI> compute_reference(const TensorShape &shape_a, const TensorShape &shape_b, const T… in compute_reference()
336 …void setup(TensorShape shape_a, TensorShape shape_b, TensorShape shape_output, int32_t a_offset, i… in setup()
|
| D | GEMMFixture.h | 56 …void setup(TensorShape shape_a, TensorShape shape_b, TensorShape shape_c, TensorShape output_shape… in setup()
86 …TensorType compute_target(const TensorShape &shape_a, const TensorShape &shape_b, const TensorShap… in compute_target()
151 …SimpleTensor<T> compute_reference(const TensorShape &shape_a, const TensorShape &shape_b, const Te… in compute_reference()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | horizontal_input_fusion.cc | 63 const Shape& shape_b) { in CompareShapeDimsFromLeftToRight()
97 Shape shape_b = GetInputShapeForMultiOutputFusion(*b); in FindAndSortFusionCandidates() local
|
| /external/tensorflow/tensorflow/lite/experimental/resource/ |
| D | resource_variable_test.cc | 100 std::vector<int> shape_b = {1}; in TEST() local
141 std::vector<int> shape_b = {2}; in TEST() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | process_broadcast_shapes.h | 94 const RuntimeShape* shape_b = in ProcessBroadcastShapes() local
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | partially_decluster_pass_test.cc | 303 Output shape_b = ops::Placeholder(s.WithOpName("shape_b"), DT_INT32, in TEST() local
355 Output shape_b = ops::Placeholder(s.WithOpName("shape_b"), DT_INT32, in TEST() local
380 Output shape_b = ops::Placeholder(s.WithOpName("shape_b"), DT_INT32, in TEST() local
|
| /external/ComputeLibrary/tests/validation/NEON/ |
| D | GEMMLowp.cpp | 57 shape_a, shape_b, shape_c, a_offset, b_offset) in DATA_TEST_CASE() argument
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/internal/ |
| D | saved_model_api_test.cc | 236 const TF_Shape* shape_b = TF_TensorSpecShape(tensor_spec_b); in TEST_P() local
|
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | spmd_expander_common.cc | 194 mlir::ArrayRef<int64_t> shape_a, mlir::ArrayRef<int64_t> shape_b, in GetBroadcastLayoutForElementWise()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 206 Shape shape_b = ShapeUtil::MakeShape(F32, {4, 3}, {true, false}); in TEST() local
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties_test.cc | 1991 const auto shape_b = properties.GetOutputProperties("b").at(0).shape(); in TEST_F() local
2063 const auto shape_b = properties.GetOutputProperties("b").at(0).shape(); in TEST_F() local
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | dynamic_ragged_shape.py | 1718 def _row_partitions_identical(shape_a, shape_b): argument
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 3068 std::vector<int> shape_b; member
3413 std::vector<int> shape_b; in TEST_P() member
|