| /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 …SimpleTensor<int32_t> compute_reference(const TensorShape &lhs_shape, const TensorShape &rhs_shape… in compute_reference()
1143 …TensorType compute_target(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const GEMMLH… in compute_target()
1200 …SimpleTensor<int32_t> compute_reference(const TensorShape &lhs_shape, const TensorShape &rhs_shape… in compute_reference()
1294 …TensorType compute_target(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const GEMMLH… in compute_target()
1350 …SimpleTensor<int32_t> compute_reference(const TensorShape &lhs_shape, const TensorShape &rhs_shape… 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 …SimpleTensor<int32_t> compute_reference(const TensorShape &lhs_shape, const TensorShape &rhs_shape… in compute_reference()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matrix_triangular_solve_op.cc | 36 const TensorShape lhs_shape = ctx->InputShape(0); in Compile() local
82 MatrixTriangularSolveOp::Broadcast(xla::XlaOp lhs, const TensorShape& lhs_shape, in Broadcast()
|
| D | xla_dot_op.cc | 49 const TensorShape lhs_shape = context->InputShape(0); in Compile() local
|
| D | xla_broadcast_helper_op.cc | 38 const TensorShape lhs_shape = context->InputShape(0); in Compile() local
|
| D | cwise_ops.cc | 37 TensorShape lhs_shape = ctx->InputShape(0); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_layout_assignment_test.cc | 65 Shape lhs_shape = ShapeUtil::MakeShapeWithLayout(F32, {12}, {0}); in TEST_F() local
100 Shape lhs_shape = ShapeUtil::MakeShapeWithLayout(F32, {12}, {0}); in TEST_F() local
184 Shape lhs_shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 12}, {0, 1}); in TEST_F() local
217 Shape lhs_shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 12}, {0, 1}); in TEST_F() local
|
| D | dot_op_emitter_internal.h | 38 Shape lhs_shape; member
|
| D | dot_op_emitter.cc | 69 Shape lhs_shape; member
555 const Shape& lhs_shape = lhs_array_.GetShape(); in Emit() local
616 const Shape& lhs_shape = lhs_array_.GetShape(); in EmitNaiveLlvmIrGemm() local
980 const Shape& lhs_shape = lhs_array_.GetShape(); in EmitCallToBatchRuntime() local
1006 const Shape& lhs_shape = lhs_array_.GetShape(); in GetMatMultDims() local
1036 const Shape& lhs_shape = lhs_array_.GetShape(); in GetBatchMatMultDims() local
1121 bool AreGemmShapes(const Shape& lhs_shape, const Shape& rhs_shape, in AreGemmShapes()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_hlo_cost_analysis.cc | 88 const Shape& lhs_shape = lhs->shape(); in GetConvolutionFlops() local
|
| D | gpu_layout_assignment.cc | 147 Shape lhs_shape = instr->operand(0)->shape(); in AddBackendConstraintsToDnnConvCustomCall() local
263 const Shape& lhs_shape = instruction->operand(0)->shape(); in AddBackendConstraints() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | bef_executable_test.cc | 30 Shape lhs_shape = in XLA_TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | batchmatmul_to_einsum.cc | 62 auto lhs_shape = lhs_type.getShape(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | batch_matmul.cc | 515 const RuntimeShape& lhs_shape, in EvalInt8Int8()
553 const RuntimeShape& lhs_shape, in EvalInt8Int32()
593 const RuntimeShape& lhs_shape, const TfLiteTensor* lhs, in EvalInt16()
620 OpData* data, const RuntimeShape& lhs_shape, in EvalQuantized()
739 RuntimeShape lhs_shape = in Eval() local
|
| /external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/components/cl/ |
| D | ClComponentElementwiseBinary.cpp | 66 const auto &lhs_shape = lhs->tensor_shape(); in validate() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | batch_dot_simplification.cc | 46 const Shape& lhs_shape = lhs->shape(); in ElideDegenerateBatchDimensionFromBatchDot() local
|
| D | dot_decomposer.cc | 48 const auto& lhs_shape = original_dot->operand(0)->shape(); in CanonicalizeDot() local
|
| D | shape_inference_test.cc | 432 Shape lhs_shape = ShapeUtil::MakeShape(F32, {10, 11, 3, 4}); in TEST_F() local
477 Shape lhs_shape = ShapeUtil::MakeShape(F32, {10, 11, 103, 4}); in TEST_F() local
523 Shape lhs_shape = ShapeUtil::MakeShape(F32, {10, 11, 3, 4}); in TEST_F() local
567 Shape lhs_shape = ShapeUtil::MakeShape(F32, {10, 11, 3, 4}); in TEST_F() local
617 Shape lhs_shape = ShapeUtil::MakeShape(F32, {60, 38, 17, 13}); in TEST_F() local
641 Shape lhs_shape; member
1595 Shape lhs_shape = ShapeUtil::MakeShape(F32, {5, 2, 11, 3}); in TEST_F() local
1621 Shape lhs_shape = ShapeUtil::MakeShape(F32, {2, 11, 3, 2}); in TEST_F() local
1642 Shape lhs_shape = ShapeUtil::MakeShape(F32, {2, 11, 3, 2}); in TEST_F() local
1686 Shape lhs_shape = ShapeUtil::MakeShape(F32, {2, 11, 3}); in TEST_F() local
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | batch_matmul.h | 28 inline void BatchMatMul(const RuntimeShape& lhs_shape, const float* lhs_data, in BatchMatMul()
115 inline void BatchMatMul(const RuntimeShape& lhs_shape, const int8_t* lhs_data, in BatchMatMul()
272 const RuntimeShape& lhs_shape, const int8_t* lhs_data, in BatchMatMul()
378 const RuntimeShape& lhs_shape, const int8_t* lhs_data, in BatchMatMul()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | batch_matmul.h | 54 inline void BatchMatMul(const RuntimeShape& lhs_shape, const Ta* lhs_data, in BatchMatMul()
109 inline void BatchMatMul(const RuntimeShape& lhs_shape, const int8_t* lhs_data, in BatchMatMul()
201 const RuntimeShape& lhs_shape, const T* lhs_data, in BatchMatMul()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_dense_binary_op_shared.cc | 102 TensorShape lhs_shape; in Compute() local
|
| D | cwise_ops_test.cc | 267 TensorShape lhs_shape; in BiasAddGrad() local
329 TensorShape lhs_shape, rhs_shape; in BcastAdd() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | legalize_tf.cc | 653 Value lhs_shape = GetShape(lhs, op->getLoc(), rewriter); in rewriteOpWithDynamicInput() local
697 auto lhs_shape = lhs.getType().cast<ShapedType>().getShape(); in matchAndRewrite() local
783 Value lhs_shape = GetShape(lhs, op->getLoc(), rewriter); in rewriteOpWithDynamicInput() local
838 auto lhs_shape = lhs.getType().cast<ShapedType>().getShape(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/core/kernels/mlir_generated/ |
| D | base_binary_ops_test.h | 44 void SetOpKernel(const std::string& op_name, const TensorShape& lhs_shape, in SetOpKernel()
69 const TensorShape& lhs_shape, in RunAndExpectResult()
96 const TensorShape& lhs_shape, in RunAndExpectInvalidArgument()
|
| /external/ComputeLibrary/tests/validation/CL/ |
| D | GEMMLowpMatrixMultiplyReshapedOnlyRHS.cpp | 140 const TensorShape lhs_shape(K, M, b_value); in validate_configuration() local
|