- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for lhs_batch_dimensions (0.28 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/convert_tf_xla_op_to_tf_op.cc
lhs_batch_idx.insert(dot_dimension_numbers.lhs_batch_dimensions().begin(), dot_dimension_numbers.lhs_batch_dimensions().end()); lhs_contract_idx.insert( dot_dimension_numbers.lhs_contracting_dimensions().begin(), dot_dimension_numbers.lhs_contracting_dimensions().end()); rhs_batch_idx.insert(dot_dimension_numbers.rhs_batch_dimensions().begin(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 13.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/replace_cast_hacks_with_tf_xla_ops.cc
const bool copy_left = true) { if (copy_left) { for (auto v : d1.lhs_batch_dimensions()) { d2.add_lhs_batch_dimensions(v); } for (auto v : d1.lhs_contracting_dimensions()) { d2.add_lhs_contracting_dimensions(v); } } else { for (auto v : d1.rhs_batch_dimensions()) { d2.add_rhs_batch_dimensions(v); } for (auto v : d1.rhs_contracting_dimensions()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 47.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/dot_general.cc
auto reshaped = rewriter.create<mhlo::ReshapeOp>(loc, result_type, matmul.getResult()); return reshaped.getResult(); } // Reshape for dynamic shaped operands. The result shape is // [lhs_batch_dimensions, lhs_out_dimensions, rhs_out_dimensions]. auto lhs_shape = rewriter.create<TFL::ShapeOp>( loc, RankedTensorType::get(static_cast<int32_t>(lhs_type.getRank()),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.2K bytes - Viewed (0)