Home
last modified time | relevance | path

Searched defs:rhs_shape (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_layout_assignment_test.cc67 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
Ddot_op_emitter_internal.h39 Shape rhs_shape; member
Ddot_op_emitter.cc58 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()
Dir_emitter.cc1105 const Shape& rhs_shape = rhs->shape(); in HandleConvolution() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dxla_dot_op.cc46 const TensorShape rhs_shape = context->InputShape(1); in Compile() local
Dxla_broadcast_helper_op.cc39 const TensorShape rhs_shape = context->InputShape(1); in Compile() local
Dxla_conv_op.cc45 const TensorShape rhs_shape = context->InputShape(1); in Compile() local
Dcwise_ops.cc35 const TensorShape rhs_shape = ctx->InputShape(1); in Compile() local
Dstrided_slice_op.cc262 const TensorShape rhs_shape = ctx->InputShape(4); in Compile() local
/external/tensorflow/tensorflow/python/kernel_tests/
Dtridiagonal_solve_op_test.py297 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/
Dir_emission_utils.cc47 bool AreValidGemmShapes(const Shape& lhs_shape, const Shape& rhs_shape, in AreValidGemmShapes()
68 const Shape& rhs_shape = dot.operand(1)->shape(); in DotImplementedAsGemm() local
Dgpu_layout_assignment.cc102 Shape rhs_shape = instr->operand(1)->shape(); in AddBackendConstraintsToDnnConvCustomCall() local
Dcudnn_conv_runner.cc330 const auto& rhs_shape = conv->operand(1)->shape(); in GetCudnnConvParams() local
Dgemm_thunk.cc296 const Shape& lhs_shape, const Shape& rhs_shape, in GemmThunk()
Dir_emitter.cc475 const Shape& rhs_shape = rhs_instruction->shape(); in HandleDot() local
/external/tensorflow/tensorflow/compiler/xla/service/
Ddot_decomposer.cc40 const Shape& rhs_shape = rhs->shape(); in DecomposeBatchDot() local
220 const auto& rhs_shape = original_dot->operand(1)->shape(); in CanonicalizeDot() local
Dshape_inference_test.cc405 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 …]
Dalgebraic_simplifier_test.cc3411 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
Dhlo_cost_analysis.cc440 const Shape& rhs_shape = rhs->shape(); in HandleConvolution() local
Dhlo_evaluator_typed_visitor.h1015 const Shape& rhs_shape = rhs->shape(); in HandleConvolution() local
/external/tensorflow/tensorflow/core/kernels/
Dcwise_ops_test.cc146 TensorShape rhs_shape; in BiasAdd() local
247 TensorShape lhs_shape, rhs_shape; in BcastAdd() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dmatrix_ops_simple_test.cc198 Shape rhs_shape = in TestImpl() local
Dxla_hlo_profile_test.cc193 Shape rhs_shape = ShapeUtil::MakeShape(F32, {m, k}); in XLA_TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc1055 const Shape& lhs_shape, const Shape& rhs_shape, in VerifyConvolution()