Home
last modified time | relevance | path

Searched defs:lhs_rank (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Ddot_decomposer.cc49 const int64_t lhs_rank = lhs_shape.rank(); in CanonicalizeDot() local
Dindexed_array_analysis.cc1049 int64_t lhs_rank = lhs->shape().rank(); in ComputeArrayForDotWithIndexedLhs() local
/external/tensorflow/tensorflow/lite/kernels/
Dbatch_matmul.cc145 const int lhs_rank = NumDimensions(lhs); in InitializeTemporaries() local
360 const int lhs_rank = NumDimensions(lhs_data); in Prepare() local
/external/tensorflow/tensorflow/compiler/tf2xla/ops/
Dxla_ops.cc276 const int32_t lhs_rank = c->Rank(lhs_shape_handle); in XlaDotShapeFunction() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Deinsum.cc511 int lhs_rank = in reshapeForBatchMatmul() local
Dlegalize_hlo.cc1677 const int lhs_rank = lhs_type.getRank(); in ConvertDot() local
1767 auto lhs_rank = dot_op.lhs().getType().cast<ShapedType>().getRank(); in ConvertDotOp() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc3312 int64_t lhs_rank = lhs_ty.getRank(); in inferReturnTypeComponents() local
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Ddot_handler.cc343 const DotConvDimsMapping& dims_mapping, int64_t lhs_rank, int64_t rhs_rank, in ComputeDimensionIndexMapping()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc699 const int64_t lhs_rank = lhs_shape->rank(); in BinaryOp() local