Search Options

Results per page
Sort
Preferred Languages
Advance

Results 21 - 30 of 49 for stream_executor_ (0.32 sec)

  1. tensorflow/compiler/mlir/tensorflow/translate/BUILD

            "@local_xla//xla/service/cpu:cpu_compiler",
            "@local_xla//xla/service/cpu:cpu_transfer_manager",
            "@local_xla//xla/stream_executor",
            "@local_xla//xla/stream_executor/host:host_platform",
            "@local_xla//xla/stream_executor/host:host_platform_id",
        ],
        alwayslink = 1,
    )
    
    cc_library(
        name = "split_into_island_per_op_pass",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 22:19:26 UTC 2024
    - 11K bytes
    - Viewed (0)
  2. tensorflow/c/kernels.h

    // OpKernelContext routines
    
    // TF_GetStream returns the SP_Stream available in ctx.
    // This function returns a stream only for devices registered using the
    // StreamExecutor C API
    // (tensorflow/c/experimental/stream_executor/stream_executor.h). It will return
    // nullptr and set error status in all other cases.
    // Experimental: this function doesn't have compatibility guarantees and subject
    // to change at any time.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jan 09 22:46:22 UTC 2024
    - 24.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tf2xla/internal/legalize_tf_to_hlo_test.cc

    #include "tensorflow/compiler/tf2xla/xla_compiler.h"
    #include "tensorflow/compiler/tf2xla/xla_helpers.h"
    #include "xla/client/client_library.h"
    #include "xla/shape.h"
    #include "xla/stream_executor/platform.h"
    #include "xla/stream_executor/platform_manager.h"
    #include "tensorflow/core/framework/tensor_shape.h"
    #include "tensorflow/core/framework/types.pb.h"
    #include "tensorflow/core/lib/monitoring/cell_reader.h"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sun Apr 14 20:29:34 UTC 2024
    - 6K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/xla_platform_info.cc

    #include "tensorflow/compiler/jit/xla_device_compiler_client.h"
    #include "xla/client/client_library.h"
    #include "xla/client/local_client.h"
    #include "xla/pjrt/pjrt_client.h"
    #include "xla/service/compiler.h"
    #include "xla/stream_executor/platform_manager.h"
    #include "tensorflow/core/framework/function.h"
    #include "tensorflow/core/framework/op_kernel.h"
    #include "tensorflow/core/framework/resource_mgr.h"
    #include "tensorflow/core/framework/types.h"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 17:23:27 UTC 2024
    - 17.4K bytes
    - Viewed (0)
  5. tensorflow/compiler/jit/xla_gpu_device.cc

    #include "tensorflow/compiler/jit/xla_platform_info.h"
    #include "tensorflow/compiler/tf2xla/layout_util.h"
    #include "tensorflow/compiler/tf2xla/xla_op_registry.h"
    #include "xla/stream_executor/gpu/gpu_init.h"
    #include "xla/stream_executor/platform_manager.h"
    #include "tensorflow/core/common_runtime/device_factory.h"
    #include "tensorflow/core/lib/core/status.h"
    
    namespace tensorflow {
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 6.6K bytes
    - Viewed (0)
  6. tensorflow/compiler/jit/kernels/BUILD

        "//tensorflow/core:lib",
        "//tensorflow/core:state_ops_op_lib",
        "//tensorflow/core/platform:stream_executor_no_cuda",
        "//tensorflow/core/profiler/lib:traceme",
        "@local_xla//xla/stream_executor/integrations:tf_allocator_adapter",
        "@com_google_absl//absl/types:optional",
    ]
    
    # Linked by tensorflow core, without registration of jit compilation passes.
    cc_library(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 22:46:36 UTC 2024
    - 3K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/legalize_tf_quant_test.cc

    #include "mlir/Pass/Pass.h"  // from @llvm-project
    #include "tensorflow/compiler/mlir/tf2xla/api/v2/legalize_tf.h"
    #include "xla/client/client_library.h"
    #include "xla/shape.h"
    #include "xla/stream_executor/platform.h"
    #include "xla/stream_executor/platform_manager.h"
    #include "tensorflow/core/framework/tensor_shape.h"
    #include "tensorflow/core/framework/types.pb.h"
    #include "tensorflow/core/protobuf/config.pb.h"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 29 18:43:55 UTC 2024
    - 7.2K bytes
    - Viewed (0)
  8. tensorflow/c/kernels.cc

    #include "absl/status/status.h"
    #include "absl/strings/str_cat.h"
    #include "tensorflow/c/c_api.h"
    #include "tensorflow/c/c_api_internal.h"
    #include "tensorflow/c/c_api_macros.h"
    #include "tensorflow/c/experimental/stream_executor/stream_executor.h"
    #include "tensorflow/c/tf_buffer.h"
    #include "tensorflow/c/tf_buffer_internal.h"
    #include "tensorflow/c/tf_datatype.h"
    #include "tensorflow/c/tf_status.h"
    #include "tensorflow/c/tf_status_helper.h"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 22:53:47 UTC 2024
    - 36K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tf2xla/transforms/BUILD

            "@local_xla//xla/mlir_hlo:convert_op_folder",
            "@local_xla//xla/mlir_hlo:mhlo_passes",
            "@local_xla//xla/mlir_hlo:type_conversion",
            "@local_xla//xla/stream_executor/tpu:c_api_conversions",
            "@local_xla//xla/stream_executor/tpu:tpu_api",
            "@local_xla//xla/translate/hlo_to_mhlo:attribute_importer",
            "@local_xla//xla/translate/mhlo_to_hlo:type_to_shape",
            "@stablehlo//:chlo_ops",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 21:49:50 UTC 2024
    - 17.4K bytes
    - Viewed (0)
  10. tensorflow/compiler/jit/get_compiler_ir.cc

    #include "xla/client/executable_build_options.h"
    #include "xla/client/local_client.h"
    #include "xla/service/hlo_graph_dumper.h"
    #include "xla/status_macros.h"
    #include "xla/stream_executor/host/host_platform_id.h"
    #include "xla/stream_executor/platform.h"
    #include "tensorflow/core/common_runtime/eager/tensor_handle.h"
    #include "tensorflow/core/framework/device_base.h"
    #include "tensorflow/core/framework/function.h"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 06:59:07 UTC 2024
    - 19K bytes
    - Viewed (0)
Back to top