Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 134 for _xla (0.04 sec)

  1. tensorflow/compiler/jit/xla_device_compiler_client.h

    #include "tensorflow/compiler/jit/device_compiler_client.h"
    #include "xla/client/local_client.h"
    
    namespace tensorflow {
    
    class XlaDeviceCompilerClient
        : public DeviceCompilerClient<xla::LocalExecutable, xla::LocalClient> {
     public:
      explicit XlaDeviceCompilerClient(xla::LocalClient* client)
          : client_(client) {}
    
      absl::StatusOr<std::unique_ptr<xla::LocalExecutable>> BuildExecutable(
          const XlaCompiler::Options& options,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 09:53:30 UTC 2024
    - 2.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/pjrt_device_compiler_client.h

    #include "tensorflow/compiler/jit/device_compiler_client.h"
    #include "xla/pjrt/pjrt_client.h"
    
    namespace tensorflow {
    
    // Calls into PjRtClient to provide functionality for building, serializing and
    // loading PjRtLoadedExecutables.
    class PjRtDeviceCompilerClient
        : public DeviceCompilerClient<xla::PjRtLoadedExecutable, xla::PjRtClient> {
     public:
      explicit PjRtDeviceCompilerClient(xla::PjRtClient* client)
          : client_(client) {}
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 12:19:41 UTC 2024
    - 3.4K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tf2xla/api/v2/legalize_tf.cc

        return absl::OkStatus();
      }
    
      TF_ASSIGN_OR_RETURN(
          auto hlo_module_config,
          xla::HloModule::CreateModuleConfigFromProto(
              compilation_result->computation->proto(), xla::DebugOptions()));
    
      TF_ASSIGN_OR_RETURN(
          std::unique_ptr<xla::HloModule> hlo_module,
          xla::HloModule::CreateFromProto(compilation_result->computation->proto(),
                                          hlo_module_config));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 29 00:40:46 UTC 2024
    - 6.8K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tf2xla/internal/legalize_tf_mlir.cc

    #include "tensorflow/compiler/tf2xla/layout_util.h"
    #include "tensorflow/compiler/tf2xla/xla_compiler.h"
    #include "tensorflow/compiler/tf2xla/xla_helpers.h"
    #include "xla/mlir_hlo/mhlo/IR/register.h"
    #include "xla/shape.h"
    #include "tensorflow/core/framework/tensor_shape.h"
    #include "tensorflow/core/platform/errors.h"
    #include "tensorflow/core/platform/profile_utils/cpu_utils.h"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sun Apr 14 20:29:34 UTC 2024
    - 6.1K bytes
    - Viewed (0)
  5. tensorflow/compiler/jit/xla_device_compiler_client.cc

    #include <memory>
    #include <string>
    #include <utility>
    #include <vector>
    
    #include "xla/client/local_client.h"
    
    namespace tensorflow {
    namespace {
    std::vector<const xla::Shape*> GetShapePointers(
        absl::Span<const xla::Shape> shapes) {
      std::vector<const xla::Shape*> shape_ptrs;
      shape_ptrs.reserve(shapes.size());
      for (const auto& shape : shapes) {
        shape_ptrs.push_back(&shape);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 09:53:30 UTC 2024
    - 4.1K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tf2xla/transforms/xla_legalize_tf_passes.td

        "quant::QuantizationDialect",
        "shape::ShapeDialect",
        "sparse_tensor::SparseTensorDialect",
        "stablehlo::StablehloDialect"
      ];
    }
    
    def LegalizeTFCollective : Pass<"xla-legalize-tf-collective", "ModuleOp"> {
      let summary = "Legalize TF/XLA collective ops (TensorFlow dialect) to the HLO dialect";
    
      let constructor = "mlir::mhlo::CreateLegalizeTFCollectivePass()";
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 21 17:44:14 UTC 2024
    - 4.3K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/stablehlo/transforms/tf_stablehlo_pass.cc

    #include "tensorflow/compiler/mlir/tf2xla/transforms/passes.h"
    #include "xla/mlir_hlo/mhlo/IR/hlo_ops.h"
    #include "xla/mlir_hlo/mhlo/IR/register.h"
    #include "xla/mlir_hlo/mhlo/transforms/passes.h"
    #include "xla/mlir_hlo/mhlo/transforms/rewriters.h"
    #include "xla/mlir_hlo/mhlo/utils/type_conversion.h"
    
    namespace mlir {
    namespace odml {
    
    class TFToMhloPass
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 21:49:50 UTC 2024
    - 7.5K bytes
    - Viewed (0)
  8. tensorflow/c/experimental/stream_executor/BUILD

            "@local_tsl//tsl/platform:status",
            "@local_xla//xla/stream_executor",
            "@local_xla//xla/stream_executor:executor_cache",
            "@local_xla//xla/stream_executor:host_memory_allocation",
            "@local_xla//xla/stream_executor:memory_allocation",
            "@local_xla//xla/stream_executor:platform",
            "@local_xla//xla/stream_executor:stream_executor_common",
        ],
    )
    
    cc_library(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jun 08 00:27:07 UTC 2024
    - 3.1K bytes
    - Viewed (0)
  9. tensorflow/compiler/jit/xla_host_send_device_context.h

    #ifndef TENSORFLOW_COMPILER_JIT_XLA_HOST_SEND_DEVICE_CONTEXT_H_
    #define TENSORFLOW_COMPILER_JIT_XLA_HOST_SEND_DEVICE_CONTEXT_H_
    
    #include "xla/shape.h"
    #include "xla/stream_executor/device_memory.h"
    #include "xla/stream_executor/stream.h"
    #include "tensorflow/core/framework/device_base.h"
    #include "tfrt/concurrency/async_value_ref.h"  // from @tf_runtime
    
    namespace tensorflow {
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 22:46:36 UTC 2024
    - 3.7K bytes
    - Viewed (0)
  10. tensorflow/compiler/jit/pjrt_device_compiler_client.cc

    #include "tensorflow/compiler/jit/pjrt_device_compiler_client.h"
    
    #include <memory>
    #include <string>
    #include <utility>
    
    namespace tensorflow {
    
    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 =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 12:19:41 UTC 2024
    - 3.6K bytes
    - Viewed (0)
Back to top