| /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 | 45 const TensorShape lhs_shape = context->InputShape(0); in Compile() local
|
| D | cwise_ops.cc | 34 const TensorShape lhs_shape = ctx->InputShape(0); in Compile() local
|
| D | xla_broadcast_helper_op.cc | 38 const TensorShape lhs_shape = context->InputShape(0); in Compile() local
|
| D | xla_conv_op.cc | 44 const TensorShape lhs_shape = context->InputShape(0); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gemm_thunk.h | 38 Shape lhs_shape; member
|
| D | gpu_layout_assignment.cc | 123 Shape lhs_shape = instr->operand(0)->shape(); in AddBackendConstraintsToDnnConvCustomCall() local
|
| D | gemm_thunk.cc | 182 const Shape &lhs_shape = gemm_config.lhs_shape; in RunGemm() local
|
| D | ir_emission_utils.cc | 55 bool AreValidGemmShapes(const Shape& lhs_shape, const Shape& rhs_shape, in AreValidGemmShapes()
110 const Shape& lhs_shape = dot.operand(0)->shape(); in IsMatrixMultiplication() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_layout_assignment_test.cc | 66 Shape lhs_shape = ShapeUtil::MakeShapeWithLayout(F32, {12}, {0}); in TEST_F() local
101 Shape lhs_shape = ShapeUtil::MakeShapeWithLayout(F32, {12}, {0}); in TEST_F() local
185 Shape lhs_shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 12}, {0, 1}); in TEST_F() local
218 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
517 const Shape& lhs_shape = lhs_array_.GetShape(); in Emit() local
552 const Shape& lhs_shape = lhs_array_.GetShape(); in EmitNaiveLlvmIrGemm() local
853 const Shape& lhs_shape = lhs_array_.GetShape(); in GetMatMultDims() local
938 bool AreGemmShapes(const Shape& lhs_shape, const Shape& rhs_shape, in AreGemmShapes()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | batchmatmul_to_einsum.cc | 61 auto lhs_shape = lhs_type.getShape(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | batch_dot_simplification.cc | 47 const Shape& lhs_shape = lhs->shape(); in ElideDegenerateBatchDimensionFromBatchDot() local
|
| D | dot_decomposer.cc | 46 const auto& lhs_shape = original_dot->operand(0)->shape(); in CanonicalizeDot() local
|
| D | shape_inference_test.cc | 406 Shape lhs_shape = ShapeUtil::MakeShape(F32, {10, 11, 3, 4}); in TEST_F() local
451 Shape lhs_shape = ShapeUtil::MakeShape(F32, {10, 11, 103, 4}); in TEST_F() local
497 Shape lhs_shape = ShapeUtil::MakeShape(F32, {10, 11, 3, 4}); in TEST_F() local
541 Shape lhs_shape = ShapeUtil::MakeShape(F32, {10, 11, 3, 4}); in TEST_F() local
591 Shape lhs_shape = ShapeUtil::MakeShape(F32, {60, 38, 17, 13}); in TEST_F() local
615 Shape lhs_shape; member
1560 Shape lhs_shape = ShapeUtil::MakeShape(F32, {5, 2, 11, 3}); in TEST_F() local
1586 Shape lhs_shape = ShapeUtil::MakeShape(F32, {2, 11, 3, 2}); in TEST_F() local
1607 Shape lhs_shape = ShapeUtil::MakeShape(F32, {2, 11, 3, 2}); in TEST_F() local
1651 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()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | batch_matmul.h | 53 inline void BatchMatMul(const RuntimeShape& lhs_shape, const float* lhs_data, in BatchMatMul()
108 inline void BatchMatMul(const RuntimeShape& lhs_shape, const int8_t* lhs_data, in BatchMatMul()
200 const RuntimeShape& lhs_shape, const T* lhs_data, in BatchMatMul()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | batch_matmul.cc | 509 const RuntimeShape& lhs_shape, const TfLiteTensor* lhs, in EvalInt8()
544 const RuntimeShape& lhs_shape, const TfLiteTensor* lhs, in EvalInt16()
571 OpData* data, const RuntimeShape& lhs_shape, in EvalQuantized()
684 RuntimeShape lhs_shape = in Eval() local
|
| /external/tensorflow/tensorflow/core/kernels/mlir_generated/ |
| D | base_binary_ops_test.h | 43 void SetOpKernel(const std::string& op_name, const TensorShape& lhs_shape, in SetOpKernel()
68 const TensorShape& lhs_shape, in RunAndExpectResult()
93 const TensorShape& lhs_shape, in RunAndExpectInvalidArgument()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cwise_ops_test.cc | 263 TensorShape lhs_shape; in BiasAddGrad() local
325 TensorShape lhs_shape, rhs_shape; in BcastAdd() local
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | lower_general_dot.cc | 165 auto lhs_shape = lhs_shape_type.getShape(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_batch_matmul_op.cc | 178 const TensorShape& lhs_shape, const TensorShape& rhs_shape, in CreateMatMulParams()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | matrix_ops_simple_test.cc | 196 Shape lhs_shape = in TestImpl() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | legalize_tf.cc | 598 auto lhs_shape = lhs.getType().cast<ShapedType>().getShape(); in matchAndRewrite() local
666 auto lhs_shape = lhs.getType().cast<ShapedType>().getShape(); in matchAndRewrite() local
|