- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 106 for _xla (0.04 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.cc
} return mlir::success(); } bool UnsupportedPartitionedShardingType(xla::OpSharding::Type sharding) { return sharding != xla::OpSharding::REPLICATED && sharding != xla::OpSharding::OTHER; } } // namespace absl::StatusOr<std::map<int, int>> GetDimensionIndicesAndNumSplitsFromSharding( const xla::OpSharding& sharding) { int64_t tensor_tile_rank = sharding.tile_assignment_dimensions_size();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 21:28:13 UTC 2024 - 34K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_metadata_utils.cc
#include "tensorflow/compiler/mlir/tensorflow/utils/convert_tensor.h" #include "tensorflow/compiler/mlir/tensorflow/utils/convert_type.h" #include "tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.h" #include "xla/xla.pb.h" #include "xla/xla_data.pb.h" #include "tensorflow/core/framework/tensor_shape.h" #include "tensorflow/core/framework/tensor_shape.pb.h" #include "tensorflow/core/framework/types.pb.h" #include "tensorflow/core/lib/core/status.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/tf2xla_rewriter_test.cc
#include "tensorflow/compiler/mlir/tf2xla/transforms/test_utils.h" #include "tensorflow/compiler/tf2xla/xla_op_registry.h" #include "xla/client/xla_builder.h" #include "xla/client/xla_computation.h" #include "xla/mlir_hlo/mhlo/IR/hlo_ops.h" #include "xla/shape_util.h" #include "xla/xla_data.pb.h" #include "tensorflow/core/framework/op_kernel.h" #include "tsl/lib/core/status_test_util.h" #include "tsl/platform/errors.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:16:07 UTC 2024 - 11.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/convert_tf_xla_op_to_tf_op.cc
// This is the argument used to refer to the pass in // the textual format (on the commandline for example). return "quant-convert-tf-xla-op-to-tf-op"; } StringRef getDescription() const final { // This is a brief description of the pass. return "Apply converting Tensorflow Xla ops to non-xla ops."; } void getDependentDialects(DialectRegistry& registry) const override {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 13.2K bytes - Viewed (0) -
tensorflow/compiler/aot/codegen_test_h.golden
::xla::cpu_function_runtime::BufferInfo(::xla::cpu_function_runtime::EncodedBufferInfo{386ULL, 1U, ~0U}), ::xla::cpu_function_runtime::BufferInfo(::xla::cpu_function_runtime::EncodedBufferInfo{5ULL, ~0U, ~0U}), ::xla::cpu_function_runtime::BufferInfo(::xla::cpu_function_runtime::EncodedBufferInfo{386ULL, 2U, ~0U}), ::xla::cpu_function_runtime::BufferInfo(::xla::cpu_function_runtime::EncodedBufferInfo{5ULL, ~0U, ~0U}),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 01:20:01 UTC 2024 - 16.6K bytes - Viewed (0) -
tensorflow/compiler/aot/codegen.cc
#include "tensorflow/compiler/tf2xla/tf2xla_util.h" #include "xla/cpu_function_runtime.h" #include "xla/service/compiler.h" #include "xla/service/cpu/buffer_info_util.h" #include "xla/shape_util.h" #include "xla/xla_data.pb.h" #include "tensorflow/core/lib/core/errors.h" namespace tensorflow { namespace tfcompile { namespace { using BufferInfo = xla::cpu_function_runtime::BufferInfo; bool IsAlpha(char c) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 01:20:01 UTC 2024 - 36.8K bytes - Viewed (0) -
tensorflow/compiler/jit/flags.cc
"Minimum number of operators in an XLA compilation. Ignored for " "operators placed on an XLA device or operators explicitly marked " "for compilation."), Flag("tf_xla_max_cluster_size", &mark_for_compilation_flags->tf_xla_max_cluster_size, "Maximum number of operators in an XLA compilation."), Flag( "tf_xla_ops_to_cluster",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 18:52:57 UTC 2024 - 24.5K bytes - Viewed (0) -
tensorflow/compiler/aot/BUILD
"@local_xla//xla:xla_data_proto_cc", "@local_xla//xla/client:client_library", "@local_xla//xla/client:compile_only_client", "@local_xla//xla/client:xla_computation", "@local_xla//xla/service:compiler", "@local_xla//xla/service/cpu:buffer_info_util", "@local_xla//xla/service/cpu:cpu_compiler", "@local_xla//xla/stream_executor:platform_manager", ], ) tf_cc_test(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 11 16:13:05 UTC 2024 - 11.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/BUILD
"@local_tsl//tsl/platform:protobuf", "@local_xla//xla/client:client_library", "@local_xla//xla/client:compile_only_client", "@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", ],
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 11K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/decompose_resource_ops.td
$resource, (TF_SubOp (CreateTFReadVariableOp $src_op, $value, $resource), $value ), (CreateConstBoolAttrFalse) ) >; // This decomposition is only correct inside XLA as it ignores use_locking // attribute. def DecomposeResourceApplyGradientDescentOp : Pat< (TF_ResourceApplyGradientDescentOp:$src_op $resource, $alpha, $delta, BoolAttr:$_), (TF_AssignVariableOp $resource,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 19:47:48 UTC 2024 - 20.7K bytes - Viewed (0)