Home
last modified time | relevance | path

Searched defs:lhs_shape (Results 1 – 25 of 46) sorted by relevance

12

/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/
H A DGEMMFixture.h263 …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 …]
H A DGEMMLowpFixture.h989 …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 …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Ddot_op_emitter.cc69 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()
H A Dcpu_layout_assignment_test.cc65 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
H A Ddot_op_emitter_internal.h38 Shape lhs_shape; member
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dbatch_matmul.cc515 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
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Dbatch_matmul.h28 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()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dmatrix_triangular_solve_op.cc36 const TensorShape lhs_shape = ctx->InputShape(0); in Compile() local
82 MatrixTriangularSolveOp::Broadcast(xla::XlaOp lhs, const TensorShape& lhs_shape, in Broadcast()
H A Dxla_dot_op.cc49 const TensorShape lhs_shape = context->InputShape(0); in Compile() local
H A Dxla_broadcast_helper_op.cc38 const TensorShape lhs_shape = context->InputShape(0); in Compile() local
H A Dcwise_ops.cc37 TensorShape lhs_shape = ctx->InputShape(0); in Compile() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dshape_inference_test.cc432 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 …]
H A Dmemory_space_assignment_test.cc6061 auto lhs_shape = ShapeUtil::MakeShape(F32, {kBatch, kFeature}); in TEST_P() local
6099 auto lhs_shape = ShapeUtil::MakeShape(F32, {kBatch, kFeature}); in TEST_P() local
6141 auto lhs_shape = ShapeUtil::MakeShape(F32, {kBatch, kFeature}); in TEST_P() local
6183 auto lhs_shape = ShapeUtil::MakeShape(F32, {kBatch, kFeature}); in TEST_P() local
6229 auto lhs_shape = ShapeUtil::MakeShape(F32, {kBatch, kFeature}); in TEST_P() local
6294 auto lhs_shape = ShapeUtil::MakeShape(F32, {kBatch, kFeature}); in TEST_P() local
6328 auto lhs_shape = ShapeUtil::MakeShape(F32, {kBatch, kFeature}); in TEST_P() local
6366 auto lhs_shape = ShapeUtil::MakeShape(F32, {kBatch, kFeature}); in TEST_P() local
6414 auto lhs_shape = ShapeUtil::MakeShape(F32, {kBatch, kFeature}); in TEST_P() local
6467 auto lhs_shape = ShapeUtil::MakeShape(F32, {kBatch, kFeature}); in TEST_P() local
[all …]
H A Dhlo_cost_analysis.cc321 int64_t HloCostAnalysis::GetDotFlops(const Shape& lhs_shape, in GetDotFlops()
583 const Shape& lhs_shape = lhs->shape(); in GetConvolutionFlops() local
592 const Shape& lhs_shape, in GetConvolutionFlops()
H A Dbatch_dot_simplification.cc46 const Shape& lhs_shape = lhs->shape(); in ElideDegenerateBatchDimensionFromBatchDot() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
H A Dlegalize_tf.cc653 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
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dbatch_matmul.h54 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()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_layout_assignment.cc147 Shape lhs_shape = instr->operand(0)->shape(); in AddBackendConstraintsToDnnConvCustomCall() local
263 const Shape& lhs_shape = instruction->operand(0)->shape(); in AddBackendConstraints() local
H A Dgpu_hlo_cost_analysis.cc88 const Shape& lhs_shape = lhs->shape(); in GetConvolutionFlops() local
H A Dmatmul_utils.cc234 const Shape& lhs_shape, absl::Span<const int64_t> lhs_batch_dims, in For()
330 const Shape& lhs_shape = gemm->operand(0)->shape(); in For() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mlir_generated/
H A Dbase_binary_ops_test.h44 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()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dcwise_ops_test.cc267 TensorShape lhs_shape; in BiasAddGrad() local
329 TensorShape lhs_shape, rhs_shape; in BcastAdd() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
H A Dbef_executable_test.cc30 Shape lhs_shape = in XLA_TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dbatchmatmul_to_einsum.cc62 auto lhs_shape = lhs_type.getShape(); in matchAndRewrite() local
/aosp_15_r20/external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/components/cl/
H A DClComponentElementwiseBinary.cpp66 const auto &lhs_shape = lhs->tensor_shape(); in validate() local

12