Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dxla_dot_op.cc45 const TensorShape lhs_shape = context->InputShape(0); in Compile() local
Dxla_broadcast_helper_op.cc38 const TensorShape lhs_shape = context->InputShape(0); in Compile() local
Dxla_conv_op.cc44 const TensorShape lhs_shape = context->InputShape(0); in Compile() local
Dcwise_ops.cc34 const TensorShape lhs_shape = ctx->InputShape(0); in Compile() local
Dextract_image_patches_op.cc102 std::vector<int64> lhs_shape(num_dims, 1); in Compile() local
Dstrided_slice_op.cc258 TensorShape lhs_shape; in Compile() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter_internal.h38 Shape lhs_shape; member
Dcpu_layout_assignment_test.cc66 Shape lhs_shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 12}, {0, 1}); in TEST_F() local
101 Shape lhs_shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 12}, {0, 1}); 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
Ddot_op_emitter.cc57 Shape lhs_shape; member
377 const Shape& lhs_shape = lhs_array_.GetShape(); in Emit() local
409 const Shape& lhs_shape = lhs_array_.GetShape(); in EmitNaiveLlvmIrGemm() local
687 const Shape& lhs_shape = lhs_array_.GetShape(); in GetMatMultDims() local
751 bool AreGemmShapes(const Shape& lhs_shape, const Shape& rhs_shape, in AreGemmShapes()
Dir_emitter.cc1104 const Shape& lhs_shape = lhs->shape(); in HandleConvolution() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dbatch_dot_simplification.cc29 const Shape& lhs_shape = lhs->shape(); in ElideDegenerateBatchDimensionFromBatchDot() local
Ddot_decomposer.cc39 const Shape& lhs_shape = lhs->shape(); in DecomposeBatchDot() local
172 const auto& lhs_shape = original_dot->operand(0)->shape(); in CanonicalizeDot() local
Dshape_inference_test.cc394 Shape lhs_shape = ShapeUtil::MakeShape(F32, {10, 11, 3, 4}); in TEST_F() local
439 Shape lhs_shape = ShapeUtil::MakeShape(F32, {10, 11, 103, 4}); in TEST_F() local
485 Shape lhs_shape = ShapeUtil::MakeShape(F32, {10, 11, 3, 4}); in TEST_F() local
529 Shape lhs_shape = ShapeUtil::MakeShape(F32, {10, 11, 3, 4}); in TEST_F() local
1049 Shape lhs_shape = ShapeUtil::MakeShape(F32, {5, 2, 11, 3}); in TEST_F() local
1074 Shape lhs_shape = ShapeUtil::MakeShape(F32, {2, 11, 3, 2}); in TEST_F() local
1094 Shape lhs_shape = ShapeUtil::MakeShape(F32, {2, 11, 3, 2}); in TEST_F() local
1115 Shape lhs_shape = ShapeUtil::MakeShape(F32, {2, 11, 3}); in TEST_F() local
1134 Shape lhs_shape = ShapeUtil::MakeShape(F32, {2, 11, 3}); in TEST_F() local
1153 Shape lhs_shape = ShapeUtil::MakeShape(F32, {2, 11, 3}); in TEST_F() local
[all …]
Dhlo_cost_analysis.cc240 const Shape& lhs_shape = dot->operand(0)->shape(); in HandleDot() local
439 const Shape& lhs_shape = lhs->shape(); in HandleConvolution() local
Dalgebraic_simplifier_test.cc3410 auto lhs_shape = root->operand(0)->operand(0)->shape(); in TEST_F() local
4336 Shape lhs_shape = ShapeUtil::MakeShape(element_type, lhs_dims); in TEST_P() local
4392 Shape lhs_shape = ShapeUtil::MakeShape(element_type, {m, k}); in TEST_P() local
4464 Shape lhs_shape = ShapeUtil::MakeShape(F32, {spec.m, spec.k}); in TEST_P() local
4539 Shape lhs_shape = ShapeUtil::MakeShape(F32, {spec.m, spec.k}); in TEST_P() local
4642 Shape lhs_shape = ShapeUtil::MakeShape(F32, {lhs_rows, lhs_cols}); in TEST_P() local
4712 Shape lhs_shape = ShapeUtil::MakeShape(F32, {lhs_rows, lhs_cols}); in TEST_P() local
Dhlo_evaluator_typed_visitor.h1014 const Shape& lhs_shape = lhs->shape(); in HandleConvolution() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emission_utils.cc47 bool AreValidGemmShapes(const Shape& lhs_shape, const Shape& rhs_shape, in AreValidGemmShapes()
67 const Shape& lhs_shape = dot.operand(0)->shape(); in DotImplementedAsGemm() local
Dgpu_layout_assignment.cc101 Shape lhs_shape = instr->operand(0)->shape(); in AddBackendConstraintsToDnnConvCustomCall() local
Dcudnn_conv_runner.cc329 const auto& lhs_shape = conv->operand(0)->shape(); in GetCudnnConvParams() local
Dgemm_thunk.cc296 const Shape& lhs_shape, const Shape& rhs_shape, in GemmThunk()
Dir_emitter.cc474 const Shape& lhs_shape = lhs_instruction->shape(); in HandleDot() local
/external/tensorflow/tensorflow/core/kernels/
Dcwise_ops_test.cc190 TensorShape lhs_shape; in BiasAddGrad() local
247 TensorShape lhs_shape, rhs_shape; in BcastAdd() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dmatrix_ops_simple_test.cc196 Shape lhs_shape = in TestImpl() local
Dxla_hlo_profile_test.cc192 Shape lhs_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()