Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for batch_dimensions (0.2 sec)

  2. github.com/tensorflow/tensorflow

    tensorflow/compiler/mlir/quantization/tensorflow/passes/convert_tf_xla_op_to_tf_op.td 

      (CreateEinsumOpFromXlaDotV2Op $lhs, $rhs, $dot, $dot_dimension_numbers),
  [(IsPrecisionEmpty $precision_config)]>;

// Only handles the case where batch_dimension is empty.
def IsXlaGatherWithoutBatch :
  Constraint<CPred<"IsXlaGatherWithoutBatch($0, $1)">>;

// Create Slice op from XlaGather op without batch dimension.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sun Dec 10 05:52:02 UTC 2023
    - 2.3K bytes
    - Viewed (0)
Back to top