- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for IsTuple (0.22 sec)
-
tensorflow/compiler/jit/xla_launch_util.cc
bool use_multiple_streams, std::shared_ptr<se::Event> definition_event) { xla::ShapeIndex output_index = input_output_alias.shape().IsTuple() ? xla::ShapeIndex({output_num}) : xla::ShapeIndex({}); CHECK(input_output_alias.shape().IsTuple() || output_num == 0); if (std::optional<xla::HloInputOutputAliasConfig::Alias> alias =
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/xla_tpu_device.cc
// If the representation is a tuple, we also must wait for the tuple index // buffers to be available on the destination host to device transfer // stream. if (xla_output->shaped_buffer().on_device_shape().IsTuple()) { TF_RETURN_IF_ERROR(dst_xla_context->host_to_device_stream()->WaitFor( dst_compute_stream)); } } for (const auto& leaf : xla_input->shaped_buffer().buffers().leaves()) {
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/mlir/tf2xla/api/v1/compile_tf_graph.cc
compilation_result->computation->GetProgramShape(); TF_RET_CHECK(program_shape_or_status.ok()); xla::ProgramShape& program_shape = program_shape_or_status.value(); if (!program_shape.result().IsTuple()) return errors::Internal("Expect result to have tuple shape"); xla::HloInputOutputAliasConfig config(program_shape.result()); for (auto alias : output_to_input_alias) { if (use_tuple_args) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 14K bytes - Viewed (0) -
src/cmd/compile/internal/ssa/schedule.go
sort.Sort(bySourcePos(order[start:])) } } return order } // isFlagOp reports if v is an OP with the flag type. func (v *Value) isFlagOp() bool { if v.Type.IsFlags() || v.Type.IsTuple() && v.Type.FieldType(1).IsFlags() { return true } // PPC64 carry generators put their carry in a non-flag-typed register // in their output. switch v.Op {
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Mon Apr 08 15:53:17 UTC 2024 - 16.4K bytes - Viewed (0) -
src/cmd/compile/internal/ssa/check.go
} } } memCheck(f) } func memCheck(f *Func) { // Check that if a tuple has a memory type, it is second. for _, b := range f.Blocks { for _, v := range b.Values { if v.Type.IsTuple() && v.Type.FieldType(0).IsMemory() { f.Fatalf("memory is first in a tuple: %s\n", v.LongString()) } } } // Single live memory checks. // These checks only work if there are no memory copies.
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Tue Apr 09 16:41:23 UTC 2024 - 17.6K bytes - Viewed (0)