- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 403 for _xla (0.04 sec)
-
tensorflow/compiler/jit/extract_outside_compilation_pass_test.cc
EXPECT_EQ(send_from_host_dtypes[0], DT_INT32); EXPECT_EQ(send_from_host_dtypes[1], DT_FLOAT); // Verify step 4: nodes marked with XLA cluster and outside compilation attr. add_node = node_name_image["add"]; EXPECT_NE(add_node, nullptr); EXPECT_TRUE(HasNodeAttr(add_node->def(), "_xla")); EXPECT_TRUE(HasNodeAttr(add_node->def(), "_oc")); // Verify step 5: control edges added. bool has_control_edge_from_recv_at_host = false;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 41K bytes - Viewed (0) -
tensorflow/compiler/jit/flags.h
// Flags associated with the XLA bridge's mark_for_compilation_pass module. struct MarkForCompilationPassFlags { XlaAutoJitFlag xla_auto_jit_flag; // Minimum number of operators in an XLA compilation. Ignored for operators // placed on an XLA device or operators explicitly marked for compilation. int32 tf_xla_min_cluster_size; // Maximum number of operators in an XLA compilation. int32 tf_xla_max_cluster_size;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 18:52:57 UTC 2024 - 14.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/quantize_composit_functions_debugging.mlir
// TF-DAG: return %[[identity]] // XLA-LABEL: func @conv_with_int_per_layer // XLA-DAG: %[[w0_float:.*]] = "tf.Const"() <{value = dense<{{\[\[\[\[}}-0.282878935, -0.211567819 // XLA-DAG: %[[b0_float:.*]] = "tf.Const"() <{value = dense<[-0.0192535277, -5.998660e-03]> : tensor<2xf32>}> : () -> tensor<2xf32> // XLA-DAG: %[[w1_float:.*]] = "tf.Const"() <{value = dense<{{\[\[\[\[}}0.208403707, 0.478067577
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Nov 06 01:23:21 UTC 2023 - 80.5K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util.cc
#include "tensorflow/compiler/tf2xla/xla_compiler.h" #include "xla/client/local_client.h" #include "xla/pjrt/pjrt_client.h" #include "xla/pjrt/pjrt_future.h" #include "xla/pjrt/pjrt_stream_executor_client.h" #include "xla/pjrt/tracked_device_buffer.h" #include "xla/shape_util.h" #include "xla/status_macros.h" #include "xla/stream_executor/platform_manager.h" #include "tensorflow/core/common_runtime/dma_helper.h"
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/jit/pjrt_device_context.cc
TF_ASSIGN_OR_RETURN(xla::Shape shape, shape_determination_fns.shape_representation_fn( cpu_tensor->shape(), cpu_tensor->dtype(), /*fast_mem=*/false, layout_preference)); const xla::Layout* device_layout = &(shape.layout()); // The device id should match the local_hardware_id in // tensorflow/compiler/xla/pjrt/pjrt_client.h. TF_ASSIGN_OR_RETURN(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 08:49:31 UTC 2024 - 11.6K bytes - Viewed (0) -
tensorflow/compiler/aot/codegen_test.cc
compile_result.program_shape = xla::ShapeUtil::MakeProgramShape( { xla::ShapeUtil::MakeShape(xla::F32, {1, 2}), xla::ShapeUtil::MakeShape(xla::S64, {3, 4}), xla::ShapeUtil::MakeShape(xla::F32, {1}), xla::ShapeUtil::MakeShape(xla::F32, {1}), xla::ShapeUtil::MakeShape(xla::S32, {5}), }, xla::ShapeUtil::MakeTupleShape({
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 01 02:13:40 UTC 2024 - 10.2K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_platform_info.cc
namespace { using XlaDeviceCompiler = DeviceCompiler<xla::LocalExecutable, xla::LocalClient>; using PjRtDeviceCompiler = DeviceCompiler<xla::PjRtLoadedExecutable, xla::PjRtClient>; using XlaDeviceExecutablePersistor = DeviceExecutablePersistor<xla::LocalExecutable, xla::LocalClient>; using PjRtDeviceExecutablePersistor = DeviceExecutablePersistor<xla::PjRtLoadedExecutable, xla::PjRtClient>; XlaDeviceCompiler* CreateXlaDeviceCompiler(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 17:23:27 UTC 2024 - 17.4K bytes - Viewed (0) -
tensorflow/compiler/jit/kernels/xla_ops.cc
}; using XlaExecutableClosure = ExecutableClosure<xla::LocalExecutable, xla::LocalClient>; using XlaExecutableClosureStore = ExecutableClosureStore<xla::LocalExecutable, xla::LocalClient>; using PjRtExecutableClosure = ExecutableClosure<xla::PjRtLoadedExecutable, xla::PjRtClient>; using PjRtExecutableClosureStore = ExecutableClosureStore<xla::PjRtLoadedExecutable, xla::PjRtClient>;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 41.4K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util.cc
#include <utility> #include "absl/status/status.h" #include "absl/status/statusor.h" #include "absl/strings/str_cat.h" #include "tensorflow/compiler/jit/pjrt_tensor_buffer_util.h" #include "xla/pjrt/c/pjrt_c_api.h" #include "xla/pjrt/pjrt_c_api_client.h" #include "xla/pjrt/pjrt_client.h" #include "tensorflow/core/framework/resource_mgr.h" #include "tensorflow/core/framework/tensor.h" #include "tensorflow/core/framework/types.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 05:48:24 UTC 2024 - 3.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_to_mhlo_int_test.cc
#include "tensorflow/compiler/tf2xla/type_util.h" #include "xla/error_spec.h" #include "xla/literal.h" #include "xla/literal_util.h" #include "xla/mlir_hlo/mhlo/IR/hlo_ops.h" #include "xla/pjrt/cpu/cpu_client.h" #include "xla/pjrt/pjrt_client.h" #include "xla/pjrt/pjrt_executable.h" #include "xla/shape.h" #include "xla/shape_util.h" #include "xla/tests/literal_test_util.h" #include "tensorflow/core/framework/tensor.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 03 01:03:21 UTC 2024 - 35.8K bytes - Viewed (0)