Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 7 of 7 for IsTuple (0.2 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. src/cmd/compile/internal/types/type.go

    			return true
    		}
    	}
    	return false
    }
    func (t *Type) IsFlags() bool   { return t == TypeFlags }
    func (t *Type) IsVoid() bool    { return t == TypeVoid }
    func (t *Type) IsTuple() bool   { return t.kind == TTUPLE }
    func (t *Type) IsResults() bool { return t.kind == TRESULTS }
    
    // IsUntyped reports whether t is an untyped type.
    func (t *Type) IsUntyped() bool {
    	if t == nil {
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Thu Apr 04 14:29:45 UTC 2024
    - 49.5K bytes
    - Viewed (0)
  7. src/cmd/compile/internal/ssa/rewrite.go

    		}
    		if v.Op == OpPhi {
    			// A Phi implies we have reached the top of the block.
    			// The memory phi, if it exists, is always
    			// the first logical store in the block.
    			continue
    		}
    		if v.Type.IsTuple() && v.Type.FieldType(1).IsMemory() {
    			// We could handle this situation however it is likely
    			// to be very rare.
    			return false
    		}
    		if v.Op.SymEffect()&SymAddr != 0 {
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Fri Jun 07 19:02:52 UTC 2024
    - 64.2K bytes
    - Viewed (0)
Back to top