- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for xla_input_shapes (0.44 sec)
-
tensorflow/compiler/jit/xla_device_compiler_client.cc
const XlaCompiler::CompilationResult& result) { VLOG(2) << "Compiling to xla::LocalExecutable."; std::vector<const xla::Shape*> argument_layouts = GetShapePointers(result.xla_input_shapes); xla::ExecutableBuildOptions build_options = GetExecutableBuildOptions( options, result, client_->default_device_ordinal()); TF_ASSIGN_OR_RETURN( auto executables,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 4.1K bytes - Viewed (0) -
tensorflow/compiler/jit/pjrt_device_compiler_client.cc
xla::CompileOptions GetPjRtCompileOptions( const XlaCompiler::Options& options, const XlaCompiler::CompilationResult& result) { xla::CompileOptions pjrt_compile_options; pjrt_compile_options.argument_layouts = result.xla_input_shapes; pjrt_compile_options.executable_build_options = GetExecutableBuildOptions(options, result, /*default_device_ordinal=*/-1); if (pjrt_compile_options.executable_build_options.num_replicas() > 1 ||
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc
} // Converts arg_shapes to xla::Shape's and store into xla_input_shapes. Status GetXlaInputShapes( mlir::ModuleOp module, llvm::ArrayRef<TensorOrResourceShape> arg_shapes, bool use_tuple_args, const XlaShapeLayoutHelpers::ShapeDeterminationFns shape_determination_fns, std::vector<xla::Shape>* xla_input_shapes) { xla_input_shapes->clear(); mlir::func::FuncOp main_func =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 17:24:39 UTC 2024 - 45.3K bytes - Viewed (0) -
tensorflow/compiler/jit/get_compiler_ir.cc
const XlaCompiler::Options& options, const bool xla_embed_ir_in_executable = false) { std::vector<const xla::Shape*> argument_layouts( result.xla_input_shapes.size()); for (int i = 0, end = result.xla_input_shapes.size(); i < end; ++i) { argument_layouts[i] = &result.xla_input_shapes[i]; } xla::ExecutableBuildOptions build_options; if (result.collective_info) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 06:59:07 UTC 2024 - 19K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util.cc
std::vector<xla::ExecutionInput> arguments; arguments.reserve(compilation_result->xla_input_shapes.size()); for (int i = 0; i < compilation_result->xla_input_shapes.size(); ++i) { int arg_num = compilation_result->input_mapping[i]; CHECK_GE(arg_num, missing_ctx_input_prefix); const xla::Shape& device_shape = compilation_result->xla_input_shapes[i]; const xla::Shape& host_shape =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 00:36:08 UTC 2024 - 40.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tf_xla_mlir_translate.cc
output << "// InputMapping {" << absl::StrJoin(compilation_result.input_mapping, ", ") << "}\n"; for (const auto& xla_input_shape : compilation_result.xla_input_shapes) output << "// XlaInputShape " << xla_input_shape.ToString() << '\n'; output << "// XlaOutputShape " << compilation_result.xla_output_shape.ToString() << '\n';
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 18.8K bytes - Viewed (0) -
tensorflow/compiler/jit/kernels/xla_ops.cc
{ tsl::profiler::TraceMe hlo_module_activity( [&] { return absl::StrCat( "Populate Inputs (", closure.compilation_result()->xla_input_shapes.size(), ")"); }, tsl::profiler::TraceMeLevel::kInfo); for (const auto& [variable_index, variable_tensor] : closure.resource_var_snapshots()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 41.4K bytes - Viewed (0)