- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for FunctionToHloArgs (0.24 sec)
-
tensorflow/compiler/mlir/tf2xla/api/v2/legalize_tf.cc
bool ShouldFallbackToGraphCompiler( const std::variant<MlirToHloArgs, FunctionToHloArgs>& computation) { if (computation.index() == 1) return true; return std::get<0>(computation).rollout_state == ConfigProto::Experimental::MLIR_BRIDGE_ROLLOUT_DISABLED; } void DumpComputationInput( const std::variant<tpu::MlirToHloArgs, tpu::FunctionToHloArgs> computation) { if (!VLOG_IS_ON(2)) { return; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 29 00:40:46 UTC 2024 - 6.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/legalize_tf.h
// compiled XLA HLO. V1 of the tf2xla uses MLIR whereas V0 does not use MLIR. // // Inputs: // computation - The MLIR module op. It currently takes in // tpu::FunctionToHloArgs but this is deprecated. arg_shapes - The shapes of // the arguments in module_op. device_type - The device type to compile for. // use_tuple_args - Pack the incoming arg shapes into a single tuple.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 07:32:57 UTC 2024 - 2.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_tf_graph.h
// converted to a Tensorflow graph. Otherwise, the graph compiler will be run. absl::Status CompileTensorflowGraphToHlo( const std::variant<tpu::MlirToHloArgs, tpu::FunctionToHloArgs>& computation, const tpu::TPUCompileMetadataProto& metadata, bool use_tuple_args, XlaShapeLayoutHelpers::ShapeDeterminationFns shape_determination_funcs, const std::vector<tensorflow::TensorShape>& arg_shapes,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 08:08:57 UTC 2024 - 2.1K bytes - Viewed (0)