- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for XlaCallModule (0.25 sec)
-
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) -
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) -
tensorflow/compiler/mlir/tensorflow/tests/shape_inference.mlir
// : (tensor<?x1024xf32>, tensor<1024x3xf32>) -> tensor<?x3xf32> // return %3 : tensor<?x3xf32> // } // // CHECK: tf.XlaCallModule // CHECK-SAME: (tensor<?x1024xf32>, tensor<1024x3xf32>) -> tensor<?x3xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 23 17:24:10 UTC 2024 - 167.4K bytes - Viewed (0) -
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) -
tensorflow/compiler/mlir/tensorflow/tests/side-effect-analysis-test.mlir
func.return // expected-remark@above {{ID: 8}} // expected-remark@above {{Sinks: {7}}} } // ----- // Tests about we create the dependency PwStreamResults `start` and `end` // within in XlaCallModule
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Dec 20 04:39:18 UTC 2023 - 129.7K bytes - Viewed (0) -
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) -
tensorflow/compiler/mlir/tensorflow/tests/tf-ops.mlir
func.return } // ----- func.func @test_xla_call_module_with_invalid_symbol() { // expected-error @below {{refers to an undefined function: @undefined_function}} "tf.XlaCallModule"() {Sout = [], device = "", dim_args_spec = [], function_list = [@undefined_function], module = "", platforms = [], version = 4 : i64} : () -> () func.return } // -----
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 23 14:40:35 UTC 2023 - 236.4K bytes - Viewed (0)