- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 11 for shape_determination_fns (0.34 sec)
-
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc
hlo_sharding, /*use_fast_memory=*/false, shape_determination_fns, shape)); return absl::OkStatus(); } // Converts arg_shapes to xla::Shape's and store into xla_input_shapes. Status GetXlaInputShapes( mlir::ModuleOp module, llvm::ArrayRef<TensorOrResourceShape> arg_shapes, bool use_tuple_args, const XlaShapeLayoutHelpers::ShapeDeterminationFns shape_determination_fns, std::vector<xla::Shape>* xla_input_shapes) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 17:24:39 UTC 2024 - 45.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.h
mlir::ModuleOp module_op, llvm::StringRef device_type, xla::XlaComputation* xla_computation, bool use_tuple_args, bool enable_op_fallback, bool return_tuple, const XlaShapeLayoutHelpers::ShapeDeterminationFns shape_determination_fns = {}, llvm::MutableArrayRef<std::unique_ptr<mlir::Pass>> custom_legalization_passes = {}, llvm::StringRef module_name = llvm::StringRef());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 17:24:39 UTC 2024 - 10.4K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_tpu_device.cc
const auto& shape_determination_fns = dst_xla_context->shape_determination_fns(); XlaLayoutPreference layout_preference = shape_determination_fns.layout_preference_fn( input->shape(), input->dtype(), std::nullopt); TF_ASSIGN_OR_RETURN(xla::Shape shape, shape_determination_fns.shape_representation_fn(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 20.9K bytes - Viewed (0) -
tensorflow/compiler/jit/pjrt_device_context.cc
xla::PjRtClient* pjrt_client, const XlaShapeLayoutHelpers::ShapeDeterminationFns shape_determination_fns) { XlaLayoutPreference layout_preference = shape_determination_fns.layout_preference_fn( cpu_tensor->shape(), cpu_tensor->dtype(), std::nullopt); TF_ASSIGN_OR_RETURN(xla::Shape shape, shape_determination_fns.shape_representation_fn(
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/jit/xla_device.cc
std::vector<XlaShapeLayoutHelpers::ShapeDeterminationFns> shape_determination_fns, PaddedShapeFn padded_shape_fn, bool use_multiple_streams) : device_ordinal_(device_ordinal), device_type_(device_type), platform_(platform), shape_determination_fns_(std::move(shape_determination_fns)), padded_shape_fn_(std::move(padded_shape_fn)),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 21:05:42 UTC 2024 - 24.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tf_xla_mlir_translate.cc
IdentityShapeRepresentationFn(); XlaShapeLayoutHelpers::ShapeDeterminationFns shape_determination_fns{ UseNoPreferenceLayoutFn(), IdentityShapeRepresentationFn()}; return PopulateResultIOInfo(module_op, arg_shapes, /*use_tuple_args=*/false, /*use_resource_updates_for_aliases=*/false, shape_determination_fns, compilation_result); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 18.8K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_device_context.cc
device_to_device_streams_(std::move(device_to_device_streams)), client_(client), transfer_manager_(client->backend().transfer_manager()), shape_determination_fns_(std::move(shape_determination_fns)), thread_pool_(thread_pool) { CHECK(host_to_device_stream_ != nullptr); CHECK(stream_ != nullptr); } void XlaDeviceContext::CopyTensorInSameDevice(const Tensor* input_tensor,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 00:36:08 UTC 2024 - 12.7K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_device.h
class Metadata { public: Metadata(int device_ordinal, se::Platform* platform, const DeviceType& device_type, std::vector<XlaShapeLayoutHelpers::ShapeDeterminationFns> shape_determination_fns, PaddedShapeFn padded_shape_fn, bool use_multiple_streams); // The index of the device on this host. int device_ordinal() const; se::Platform* platform() const;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 13.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_tf_graph_test.cc
} XlaShapeLayoutHelpers::ShapeDeterminationFns shape_determination_fns; absl::Status compilation_status = tensorflow::tf2xla::v1::CompileTensorflowGraphToHlo( computation, metadata_proto, use_tuple_args, shape_determination_fns, arg_shapes, &arg_core_mapping, &per_core_arg_shapes, client, &compilation_result);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 08:08:57 UTC 2024 - 11.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/legalize_tf_test.cc
return LegalizeMlirToHlo(mlir_to_hlo_args, metadata_proto, use_tuple_args, /*device_type=*/"XLA_TPU_JIT", custom_legalization_passes, /*shape_determination_fns=*/{}, arg_shapes, &arg_core_mapping, &per_core_arg_shapes, client); } TEST(LegalizeTFTest, RecordsStreamzForSuccessfulLegalizeWithMlirBridge) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 23:59:33 UTC 2024 - 16.1K bytes - Viewed (0)