| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | GEMMFixture.h | 263 …TensorType compute_target(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const Tensor… in compute_target()
316 …SimpleTensor<T> compute_reference(const TensorShape &lhs_shape, const TensorShape &rhs_shape, Data… in compute_reference()
386 …TensorType compute_target(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const Tensor… in compute_target()
439 …SimpleTensor<T> compute_reference(const TensorShape &lhs_shape, const TensorShape &rhs_shape, Data… in compute_reference()
523 …TensorType compute_target(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const Tensor… in compute_target()
594 …SimpleTensor<T> compute_reference(const TensorShape &lhs_shape, const TensorShape &rhs_shape, Data… in compute_reference()
678 …TensorType compute_target(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const Tensor… in compute_target()
749 …SimpleTensor<T> compute_reference(const TensorShape &lhs_shape, const TensorShape &rhs_shape, Data… in compute_reference()
837 …TensorType compute_target(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const Tensor… in compute_target()
924 …SimpleTensor<T> compute_reference(const TensorShape &lhs_shape, const TensorShape &rhs_shape, Data… in compute_reference()
[all …]
|
| D | GEMMLowpFixture.h | 989 …TensorType compute_target(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const GEMMLH… in compute_target()
1045 …nsor<int32_t> compute_reference(const TensorShape &lhs_shape, const TensorShape &rhs_shape, DataTy… in compute_reference()
1143 …TensorType compute_target(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const GEMMLH… in compute_target()
1200 …nsor<int32_t> compute_reference(const TensorShape &lhs_shape, const TensorShape &rhs_shape, unsign… in compute_reference()
1294 …TensorType compute_target(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const GEMMLH… in compute_target()
1350 …nsor<int32_t> compute_reference(const TensorShape &lhs_shape, const TensorShape &rhs_shape, DataTy… in compute_reference()
1465 …TensorType compute_target(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const Tensor… in compute_target()
1564 …SimpleTensor<T> compute_reference(const TensorShape &lhs_shape, const TensorShape &rhs_shape, cons… in compute_reference()
1648 …TensorType compute_target(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const GEMMLH… in compute_target()
1711 …nsor<int32_t> compute_reference(const TensorShape &lhs_shape, const TensorShape &rhs_shape, DataTy… in compute_reference()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matrix_triangular_solve_op.cc | 37 const TensorShape rhs_shape = ctx->InputShape(1); in Compile() local
83 xla::XlaOp rhs, const TensorShape& rhs_shape, in Broadcast()
|
| D | xla_dot_op.cc | 50 const TensorShape rhs_shape = context->InputShape(1); in Compile() local
|
| D | xla_broadcast_helper_op.cc | 39 const TensorShape rhs_shape = context->InputShape(1); in Compile() local
|
| D | cwise_ops.cc | 38 TensorShape rhs_shape = ctx->InputShape(1); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_layout_assignment_test.cc | 66 Shape rhs_shape = ShapeUtil::MakeShape(F32, {12, 24}); in TEST_F() local
101 Shape rhs_shape = ShapeUtil::MakeShape(F32, {12, 24}); in TEST_F() local
144 Shape rhs_shape = ShapeUtil::MakeShapeWithLayout(F32, {12, 24}, {0, 1}); in TEST_F() local
185 Shape rhs_shape = ShapeUtil::MakeShapeWithLayout(F32, {12, 24}, {0, 1}); in TEST_F() local
218 Shape rhs_shape = ShapeUtil::MakeShapeWithLayout(F32, {12, 24}, {0, 1}); in TEST_F() local
|
| D | dot_op_emitter_internal.h | 39 Shape rhs_shape; member
|
| D | dot_op_emitter.cc | 70 Shape rhs_shape; member
556 const Shape& rhs_shape = rhs_array_.GetShape(); in Emit() local
617 const Shape& rhs_shape = rhs_array_.GetShape(); in EmitNaiveLlvmIrGemm() local
1007 const Shape& rhs_shape = rhs_array_.GetShape(); in GetMatMultDims() local
1037 const Shape& rhs_shape = rhs_array_.GetShape(); in GetBatchMatMultDims() local
1121 bool AreGemmShapes(const Shape& lhs_shape, const Shape& rhs_shape, in AreGemmShapes()
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| D | tridiagonal_solve_op_test.py | 509 def test_raises(diags_shape, rhs_shape): argument
520 def test_raises(diags_tuple_shapes, rhs_shape): argument
534 def test_raises(diags_shape, rhs_shape): argument
545 rhs_shape, argument
604 def test_with_matrix_shapes(matrix_shape, rhs_shape=None): argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_hlo_cost_analysis.cc | 89 const Shape& rhs_shape = rhs->shape(); in GetConvolutionFlops() local
|
| D | gpu_layout_assignment.cc | 148 Shape rhs_shape = instr->operand(1)->shape(); in AddBackendConstraintsToDnnConvCustomCall() local
264 const Shape& rhs_shape = instruction->operand(1)->shape(); in AddBackendConstraints() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | tridiagonal_solve_ops_test.py | 473 def test_raises(diags_shape, rhs_shape): argument
483 def test_raises(diags_tuple_shapes, rhs_shape): argument
496 def test_raises(diags_shape, rhs_shape): argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | bef_executable_test.cc | 32 Shape rhs_shape = in XLA_TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | batchmatmul_to_einsum.cc | 63 auto rhs_shape = rhs_type.getShape(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | batch_matmul.cc | 517 const RuntimeShape& rhs_shape, in EvalInt8Int8()
555 const RuntimeShape& rhs_shape, in EvalInt8Int32()
594 const RuntimeShape& rhs_shape, const TfLiteTensor* rhs, in EvalInt16()
622 const RuntimeShape& rhs_shape, in EvalQuantized()
737 RuntimeShape rhs_shape = in Eval() local
|
| /external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/components/cl/ |
| D | ClComponentElementwiseBinary.cpp | 67 const auto &rhs_shape = rhs->tensor_shape(); in validate() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | batch_matmul.h | 29 const RuntimeShape& rhs_shape, const float* rhs_data, in BatchMatMul()
116 const RuntimeShape& rhs_shape, const int8_t* rhs_data, in BatchMatMul()
273 const RuntimeShape& rhs_shape, const int8_t* rhs_data, in BatchMatMul()
379 const RuntimeShape& rhs_shape, const int8_t* rhs_data, in BatchMatMul()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | batch_matmul.h | 55 const RuntimeShape& rhs_shape, const Tb* rhs_data, in BatchMatMul()
110 const RuntimeShape& rhs_shape, const int8_t* rhs_data, in BatchMatMul()
202 const RuntimeShape& rhs_shape, const T* rhs_data, in BatchMatMul()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | legalize_tf.cc | 654 Value rhs_shape = GetShape(rhs, op->getLoc(), rewriter); in rewriteOpWithDynamicInput() local
698 auto rhs_shape = rhs.getType().cast<ShapedType>().getShape(); in matchAndRewrite() local
784 Value rhs_shape = GetShape(rhs, op->getLoc(), rewriter); in rewriteOpWithDynamicInput() local
839 auto rhs_shape = rhs.getType().cast<ShapedType>().getShape(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cwise_ops_test.cc | 218 TensorShape rhs_shape; in BiasAdd() local
329 TensorShape lhs_shape, rhs_shape; in BcastAdd() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dot_decomposer.cc | 102 const auto& rhs_shape = original_dot->operand(1)->shape(); in CanonicalizeDot() local
|
| D | shape_inference_test.cc | 443 Shape rhs_shape = ShapeUtil::MakeShape(F32, {2, 12, 11, 3}); in TEST_F() local
488 Shape rhs_shape = ShapeUtil::MakeShape(F32, {2, 12, 11, 3}); in TEST_F() local
534 Shape rhs_shape = ShapeUtil::MakeShape(F32, {2, 12, 11, 4}); in TEST_F() local
568 Shape rhs_shape = ShapeUtil::MakeShape(F32, {12, 11, 3, 2}); in TEST_F() local
618 Shape rhs_shape = ShapeUtil::MakeShape(F32, {38, 10, 4, 4}); in TEST_F() local
642 Shape rhs_shape; member
1596 Shape rhs_shape = ShapeUtil::MakeShape(F32, {5, 2, 3, 14}); in TEST_F() local
1622 Shape rhs_shape = ShapeUtil::MakeShape(F32, {2, 3, 14}); in TEST_F() local
1643 Shape rhs_shape = ShapeUtil::MakeShape(F32, {2, 3, 2, 14}); in TEST_F() local
1687 Shape rhs_shape = ShapeUtil::MakeShape(F32, {3, 3, 14}); in TEST_F() local
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/mlir_generated/ |
| D | base_binary_ops_test.h | 46 const TensorShape& rhs_shape, in SetOpKernel()
71 const TensorShape& rhs_shape, in RunAndExpectResult()
98 const TensorShape& rhs_shape, in RunAndExpectInvalidArgument()
|
| /external/ComputeLibrary/tests/validation/CL/ |
| D | GEMMLowpMatrixMultiplyReshapedOnlyRHS.cpp | 141 const TensorShape rhs_shape(N, K, b_value); in validate_configuration() local
|