Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for XlaCallModule (1.57 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td

      let summary = "Serializes StableHLO functions from top-level module into `tf.XlaCallModule`'s `module` attribute";
    
      let description = [{
        This pass collects StableHLO functions referenced from `tf.XlaCallModule`'s
        `_entry_function` attribute into a module, serializes the module into MLIR
        bytecode, and embed the bytecode to `tf.XlaCallModule`'s `module` attribute.
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:18:05 UTC 2024
    - 99.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc

      // TODO(b/154065712): Remove propagate_caller_callee_constants once using
      // SCCP pass instead.
      bool propagate_caller_callee_constants_;
    
      // XlaCallModule loader, which is used to deserialize the StableHLO module in
      // each `XlaCallModule` op. Uses its own MLIRContext since the loader needs to
      // load additional dialects, which is not allowed for the main context since
      // shape inference may be called from a pass.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jun 08 07:28:49 UTC 2024
    - 134.1K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test.py

          # This is to verify the invocation of StableHLO quantizer works. More
          # thorough functional tests are in StableHLO quantizer directory.
          self.assertTrue(self._contains_op(output_graphdef, 'XlaCallModule'))
        else:
          self.assertTrue(self._contains_quantized_function_call(output_graphdef))
        self.assertFalse(self._contains_op(output_graphdef, 'FusedBatchNormV3'))
    
      @parameterized.named_parameters(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 03:36:50 UTC 2024
    - 235.6K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

    void XlaCallModuleOp::getEffects(
        SmallVectorImpl<SideEffects::EffectInstance<MemoryEffects::Effect>>
            &effects) {
      if (!getFunctionList().empty()) {
        // The StableHLO module embedded in XlaCallModule contains
        // `stablehlo.custom_call` calling TF host callback functions.
        // `stablehlo.custom_call` will be lowered to `stablehlo.send` and
        // `stablehlo.recv`.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
Back to top