| /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 | 46 const TensorShape rhs_shape = context->InputShape(1); in Compile() local
|
| D | cwise_ops.cc | 35 const TensorShape rhs_shape = ctx->InputShape(1); in Compile() local
|
| D | xla_broadcast_helper_op.cc | 39 const TensorShape rhs_shape = context->InputShape(1); in Compile() local
|
| D | xla_conv_op.cc | 45 const TensorShape rhs_shape = context->InputShape(1); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_layout_assignment_test.cc | 67 Shape rhs_shape = ShapeUtil::MakeShape(F32, {12, 24}); in TEST_F() local
102 Shape rhs_shape = ShapeUtil::MakeShape(F32, {12, 24}); in TEST_F() local
145 Shape rhs_shape = ShapeUtil::MakeShapeWithLayout(F32, {12, 24}, {0, 1}); in TEST_F() local
186 Shape rhs_shape = ShapeUtil::MakeShapeWithLayout(F32, {12, 24}, {0, 1}); in TEST_F() local
219 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
518 const Shape& rhs_shape = rhs_array_.GetShape(); in Emit() local
553 const Shape& rhs_shape = rhs_array_.GetShape(); in EmitNaiveLlvmIrGemm() local
854 const Shape& rhs_shape = rhs_array_.GetShape(); in GetMatMultDims() local
938 bool AreGemmShapes(const Shape& lhs_shape, const Shape& rhs_shape, in AreGemmShapes()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gemm_thunk.h | 39 Shape rhs_shape; member
|
| D | gpu_layout_assignment.cc | 124 Shape rhs_shape = instr->operand(1)->shape(); in AddBackendConstraintsToDnnConvCustomCall() local
|
| D | gemm_thunk.cc | 183 const Shape &rhs_shape = gemm_config.rhs_shape; in RunGemm() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | tridiagonal_solve_op_test.py | 516 def test_raises(diags_shape, rhs_shape): argument
527 def test_raises(diags_tuple_shapes, rhs_shape): argument
541 def test_raises(diags_shape, rhs_shape): argument
552 rhs_shape, argument
611 def test_with_matrix_shapes(matrix_shape, rhs_shape=None): argument
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | tridiagonal_solve_ops_test.py | 477 def test_raises(diags_shape, rhs_shape): argument
487 def test_raises(diags_tuple_shapes, rhs_shape): argument
500 def test_raises(diags_shape, rhs_shape): argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | batchmatmul_to_einsum.cc | 62 auto rhs_shape = rhs_type.getShape(); in matchAndRewrite() 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()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | batch_matmul.h | 54 const RuntimeShape& rhs_shape, const float* rhs_data, in BatchMatMul()
109 const RuntimeShape& rhs_shape, const int8_t* rhs_data, in BatchMatMul()
201 const RuntimeShape& rhs_shape, const T* rhs_data, in BatchMatMul()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | batch_matmul.cc | 510 const RuntimeShape& rhs_shape, const TfLiteTensor* rhs, in EvalInt8()
545 const RuntimeShape& rhs_shape, const TfLiteTensor* rhs, in EvalInt16()
573 const RuntimeShape& rhs_shape, in EvalQuantized()
682 RuntimeShape rhs_shape = in Eval() local
|
| /external/tensorflow/tensorflow/core/kernels/mlir_generated/ |
| D | base_binary_ops_test.h | 45 const TensorShape& rhs_shape, in SetOpKernel()
70 const TensorShape& rhs_shape, in RunAndExpectResult()
95 const TensorShape& rhs_shape, in RunAndExpectInvalidArgument()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dot_decomposer.cc | 96 const auto& rhs_shape = original_dot->operand(1)->shape(); in CanonicalizeDot() local
|
| D | shape_inference_test.cc | 417 Shape rhs_shape = ShapeUtil::MakeShape(F32, {2, 12, 11, 3}); in TEST_F() local
462 Shape rhs_shape = ShapeUtil::MakeShape(F32, {2, 12, 11, 3}); in TEST_F() local
508 Shape rhs_shape = ShapeUtil::MakeShape(F32, {2, 12, 11, 4}); in TEST_F() local
542 Shape rhs_shape = ShapeUtil::MakeShape(F32, {12, 11, 3, 2}); in TEST_F() local
592 Shape rhs_shape = ShapeUtil::MakeShape(F32, {38, 10, 4, 4}); in TEST_F() local
616 Shape rhs_shape; member
1561 Shape rhs_shape = ShapeUtil::MakeShape(F32, {5, 2, 3, 14}); in TEST_F() local
1587 Shape rhs_shape = ShapeUtil::MakeShape(F32, {2, 3, 14}); in TEST_F() local
1608 Shape rhs_shape = ShapeUtil::MakeShape(F32, {2, 3, 2, 14}); in TEST_F() local
1652 Shape rhs_shape = ShapeUtil::MakeShape(F32, {3, 3, 14}); in TEST_F() local
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cwise_ops_test.cc | 215 TensorShape rhs_shape; in BiasAdd() 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 | 166 auto rhs_shape = rhs_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 | 198 Shape rhs_shape = in TestImpl() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | legalize_tf.cc | 599 auto rhs_shape = rhs.getType().cast<ShapedType>().getShape(); in matchAndRewrite() local
667 auto rhs_shape = rhs.getType().cast<ShapedType>().getShape(); in matchAndRewrite() local
|