Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 9 of 9 for graph_def_version (0.32 sec)

  1. tensorflow/compiler/jit/xla_compiler_options_util.cc

              << ",device_ordinal=" << options.device_ordinal
              << ",client=" << options.client << ",flib_def=" << options.flib_def
              << ",graph_def_version=" << options.graph_def_version
              << ",options.shape_determination_fns.layout_preference_fn?="
              << (options.shape_determination_fns.layout_preference_fn != nullptr)
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 09:53:30 UTC 2024
    - 6.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/xla_compiler_options_util_test.cc

          /*pjrt_device_compiler=*/nullptr);
    
      EXPECT_EQ(options.device_type, compilation_device_type);
      EXPECT_EQ(options.device_ordinal, 0);
      EXPECT_NE(options.flib_def, nullptr);
      EXPECT_EQ(options.graph_def_version, TF_GRAPH_DEF_VERSION);
      EXPECT_FALSE(options.allow_cpu_custom_calls);
      EXPECT_FALSE(options.alias_passthrough_params);
      // Check if options have the supplied shape determination functions set.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Dec 29 01:41:20 UTC 2023
    - 14.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/xla_compiler_options_util.h

    // compilation and execution.
    XlaCompiler::Options GenerateCompilerOptionsForPjRt(
        const FunctionLibraryDefinition* function_library_def,
        int graph_def_version, const DeviceBase* device_base,
        const XlaPlatformInfo& platform_info,
        const DeviceCompiler<xla::PjRtLoadedExecutable, xla::PjRtClient>*
            pjrt_device_compiler);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Dec 29 01:41:20 UTC 2023
    - 2.7K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/xla_kernel_creator.cc

                                        flr, dev->resource_manager(), props,
                                        input_memory_types, output_memory_types,
                                        flr->graph_def_version(), &s);
    
      *kernel = std::make_unique<XlaLocalLaunchBase>(
          &construction, constant_arg_indices, resource_arg_indices, function,
          /*has_ref_vars=*/false);
      return s;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Oct 31 22:24:01 UTC 2023
    - 4.4K bytes
    - Viewed (0)
  5. tensorflow/compiler/jit/get_compiler_ir.cc

      } else {
        options.device_type = compilation_device_type;
        options.flib_def = flr->GetFunctionLibraryDefinition();
        options.graph_def_version = flr->graph_def_version();
        options.allow_cpu_custom_calls =
            (platform_info.platform_id() == se::host::kHostPlatformId);
        options.alias_passthrough_params = !platform_info.is_on_xla_device();
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 06:59:07 UTC 2024
    - 19K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tf2xla/api/v1/compile_tf_graph_test.cc

      NameAttrList function;
      function.set_name("empty");
    
      FunctionToHloArgs function_to_hlo_args = {&function,
                                                &flib_def,
                                                /*graph_def_version=*/0,
                                                {&guaranteed_constants}};
    
      TF_EXPECT_OK(CompileWithComputation(function_to_hlo_args).status());
    
      Histogram histogram =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 13 08:08:57 UTC 2024
    - 11.7K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tf2xla/api/v2/legalize_tf_test.cc

      OpInputList guaranteed_constants;
      NameAttrList function;
      FunctionToHloArgs function_to_hlo_args{&function,
                                             &flib_def,
                                             /*graph_def_version=*/0,
                                             {&guaranteed_constants}};
    
      se::Platform* cpu_platform =
          se::PlatformManager::PlatformWithName("Host").value();
      auto client =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 13 23:59:33 UTC 2024
    - 16.1K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tf2xla/api/v1/compile_tf_graph.cc

        xla::CompileOnlyClient* client,
        XlaCompiler::CompilationResult* compilation_result) {
      Status comp_status = CompileTFFunctionToHlo(
          *function_computation.flib_def, function_computation.graph_def_version,
          shape_determination_funcs, arg_shapes,
          function_computation.guaranteed_constants, *function_computation.function,
          metadata, client, arg_core_mapping, per_core_arg_shapes, use_tuple_args,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 22:19:26 UTC 2024
    - 14K bytes
    - Viewed (0)
  9. tensorflow/c/kernels.cc

          ->frame_iter()
          .frame_id;
    }
    
    int TF_GetGraphDefVersion(TF_OpKernelContext* ctx) {
      return reinterpret_cast<::tensorflow::OpKernelContext*>(ctx)
          ->function_library()
          ->graph_def_version();
    }
    
    int64_t TF_GetIterId(TF_OpKernelContext* ctx) {
      return reinterpret_cast<::tensorflow::OpKernelContext*>(ctx)
          ->frame_iter()
          .iter_id;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 22:53:47 UTC 2024
    - 36K bytes
    - Viewed (0)
Back to top