/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 | 58 Shape rhs_shape; member 378 const Shape& rhs_shape = rhs_array_.GetShape(); in Emit() local 410 const Shape& rhs_shape = rhs_array_.GetShape(); in EmitNaiveLlvmIrGemm() local 688 const Shape& rhs_shape = rhs_array_.GetShape(); in GetMatMultDims() local 751 bool AreGemmShapes(const Shape& lhs_shape, const Shape& rhs_shape, in AreGemmShapes()
|
D | ir_emitter.cc | 1105 const Shape& rhs_shape = rhs->shape(); in HandleConvolution() local
|
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | xla_dot_op.cc | 46 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 | xla_conv_op.cc | 45 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 | strided_slice_op.cc | 262 const TensorShape rhs_shape = ctx->InputShape(4); in Compile() local
|
/external/tensorflow/tensorflow/python/kernel_tests/ |
D | tridiagonal_solve_op_test.py | 297 def test_raises(diags_shape, rhs_shape): argument 307 def test_raises(diags_tuple_shapes, rhs_shape): argument 320 def test_raises(diags_shape, rhs_shape): argument 331 rhs_shape, argument
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | ir_emission_utils.cc | 47 bool AreValidGemmShapes(const Shape& lhs_shape, const Shape& rhs_shape, in AreValidGemmShapes() 68 const Shape& rhs_shape = dot.operand(1)->shape(); in DotImplementedAsGemm() local
|
D | gpu_layout_assignment.cc | 102 Shape rhs_shape = instr->operand(1)->shape(); in AddBackendConstraintsToDnnConvCustomCall() local
|
D | cudnn_conv_runner.cc | 330 const auto& rhs_shape = conv->operand(1)->shape(); in GetCudnnConvParams() local
|
D | gemm_thunk.cc | 296 const Shape& lhs_shape, const Shape& rhs_shape, in GemmThunk()
|
D | ir_emitter.cc | 475 const Shape& rhs_shape = rhs_instruction->shape(); in HandleDot() local
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | dot_decomposer.cc | 40 const Shape& rhs_shape = rhs->shape(); in DecomposeBatchDot() local 220 const auto& rhs_shape = original_dot->operand(1)->shape(); in CanonicalizeDot() local
|
D | shape_inference_test.cc | 405 Shape rhs_shape = ShapeUtil::MakeShape(F32, {2, 12, 11, 3}); in TEST_F() local 450 Shape rhs_shape = ShapeUtil::MakeShape(F32, {2, 12, 11, 3}); in TEST_F() local 496 Shape rhs_shape = ShapeUtil::MakeShape(F32, {2, 12, 11, 4}); in TEST_F() local 530 Shape rhs_shape = ShapeUtil::MakeShape(F32, {12, 11, 3, 2}); in TEST_F() local 1050 Shape rhs_shape = ShapeUtil::MakeShape(F32, {5, 2, 3, 14}); in TEST_F() local 1075 Shape rhs_shape = ShapeUtil::MakeShape(F32, {2, 3, 14}); in TEST_F() local 1095 Shape rhs_shape = ShapeUtil::MakeShape(F32, {2, 3, 2, 14}); in TEST_F() local 1116 Shape rhs_shape = ShapeUtil::MakeShape(F32, {3, 3, 14}); in TEST_F() local 1135 Shape rhs_shape = ShapeUtil::MakeShape(F32, {3, 2, 14}); in TEST_F() local 1154 Shape rhs_shape = ShapeUtil::MakeShape(F32, {2, 3, 14}); in TEST_F() local [all …]
|
D | algebraic_simplifier_test.cc | 3411 auto rhs_shape = root->operand(0)->operand(1)->shape(); in TEST_F() local 4337 Shape rhs_shape = ShapeUtil::MakeShape(element_type, rhs_dims); in TEST_P() local 4394 Shape rhs_shape = ShapeUtil::MakeShape(element_type, {k, n}); in TEST_P() local 4480 Shape rhs_shape = ShapeUtil::MakeShape(F32, {spec.k, spec.n}); in TEST_P() local 4544 Shape rhs_shape = ShapeUtil::MakeShape(F32, {spec.k, spec.n}); in TEST_P() local 4664 Shape rhs_shape = ShapeUtil::MakeShape(F32, {rhs_rows, rhs_cols}); in TEST_P() local 4723 Shape rhs_shape = ShapeUtil::MakeShape(F32, {rhs_rows, rhs_cols}); in TEST_P() local
|
D | hlo_cost_analysis.cc | 440 const Shape& rhs_shape = rhs->shape(); in HandleConvolution() local
|
D | hlo_evaluator_typed_visitor.h | 1015 const Shape& rhs_shape = rhs->shape(); in HandleConvolution() local
|
/external/tensorflow/tensorflow/core/kernels/ |
D | cwise_ops_test.cc | 146 TensorShape rhs_shape; in BiasAdd() local 247 TensorShape lhs_shape, rhs_shape; in BcastAdd() local
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | matrix_ops_simple_test.cc | 198 Shape rhs_shape = in TestImpl() local
|
D | xla_hlo_profile_test.cc | 193 Shape rhs_shape = ShapeUtil::MakeShape(F32, {m, k}); in XLA_TEST_F() local
|
/external/tensorflow/tensorflow/compiler/xla/client/ |
D | xla_builder.cc | 1055 const Shape& lhs_shape, const Shape& rhs_shape, in VerifyConvolution()
|