Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for SelectV2Op (0.09 sec)

  1. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

                                 "rank as t and e OR is a vector";
    }
    
    //===----------------------------------------------------------------------===//
    // SelectV2Op
    //===----------------------------------------------------------------------===//
    
    static Type InferSelectV2OpType(Value condition, Value e, Value t) {
      Type element_ty = e.getType().cast<TensorType>().getElementType();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc

                   PatternRewriter& rewriter) const override {
        Value pred = op.getOperand(0);
        Value on_true = op.getOperand(1);
        Value on_false = op.getOperand(2);
        rewriter.replaceOpWithNewOp<TFL::SelectV2Op>(op, pred, on_true, on_false);
      }
    };
    
    // Rewrites quantized stablehlo.concatenate to tfl.concatenation.
    // TODO: b/322428814 - Add StableHLO quantizer integration tests for ODML.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 09:00:19 UTC 2024
    - 99.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/ir/tfl_ops.cc

      };
    
      return elements_attr.mapValues(result_element_type, cast);
    }
    
    //===----------------------------------------------------------------------===//
    // SelectV2Op
    //===----------------------------------------------------------------------===//
    
    static void BuildSelectV2Op(Builder* builder, OperationState& result,
                                Value cond, Value x, Value y) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 169.2K bytes
    - Viewed (0)
Back to top