HomeSort by relevance Sort by last modified time
    Searched refs:IsTuple (Results 1 - 25 of 88) sorted by null

1 2 3 4

  /external/tensorflow/tensorflow/compiler/xla/
shape_layout.cc 55 CHECK(!shape_.IsTuple());
64 CHECK(!shape_.IsTuple());
72 CHECK(shape_.IsTuple());
layout_util.cc 143 if (shape->IsTuple()) {
175 if (shape.IsTuple()) {
207 if (shape.IsTuple()) {
307 if (shape.IsTuple()) {
377 if (src.IsTuple() != dst->IsTuple()) {
381 if (src.IsTuple()) {
415 if (lhs.IsTuple()) {
416 if (!rhs.IsTuple() || ShapeUtil::TupleElementCount(lhs) !=
shape_util.cc 329 return shape.IsTuple() &&
331 [](const Shape& s) { return s.IsTuple(); });
335 return shape.IsTuple() && TupleElementCount(shape) == 0;
339 CHECK(shape.IsTuple()) << HumanString(shape);
345 CHECK(shape.IsTuple());
361 CHECK(tuple.IsTuple());
390 CHECK(shape.IsArray() || shape.IsTuple());
424 if (shape.IsTuple()) {
448 if (shape.IsTuple()) {
773 if (!subshape->IsTuple() || i >= subshape->tuple_shapes_size() || i < 0)
    [all...]
shape.cc 87 if (IsTuple()) {
120 if (lhs.IsTuple()) {
121 return rhs.IsTuple() &&
  /external/tensorflow/tensorflow/compiler/xla/service/
optimize_input_output_buffer_alias.cc 34 return shape.IsTuple() && !ShapeUtil::IsNestedTuple(shape);
53 if (subshape.IsTuple()) {
64 if (subshape.IsTuple()) {
hlo_sharding.cc 61 CHECK(tuple_shape.IsTuple()) << ShapeUtil::HumanString(tuple_shape);
63 CHECK(!sharding.IsTuple()) << sharding.ToString();
74 CHECK(tuple_shape.IsTuple()) << ShapeUtil::HumanString(tuple_shape);
75 CHECK(!sharding.IsTuple()) << sharding.ToString();
84 return shape.IsTuple() ? SingleTuple(shape, sharding) : sharding;
88 if (IsTuple()) {
109 if (IsTuple()) {
121 if (IsTuple()) {
143 CHECK(!IsTuple());
156 CHECK(!IsTuple());
    [all...]
tuple_util.cc 24 CHECK(input_tuple->shape().IsTuple());
44 CHECK(input_tuple->shape().IsTuple());
hlo_sharding.h 97 bool IsTuple() const { return tuple_; }
101 if (!IsTuple()) {
110 if (!IsTuple()) {
130 // REQUIRES: !IsTuple()
135 // REQUIRES: !IsTuple()
141 // REQUIRES: !IsTuple()
148 // REQUIRES: !IsTuple()
164 // tuple, if IsTuple, or a ShapeTree with a single element containing this
173 // REQUIRES: IsTuple()
204 // REQUIRES: !IsReplicated() && !IsTuple()
    [all...]
buffer_value.cc 33 is_tuple_ = shape.IsTuple();
buffer_value.h 136 bool IsTuple() const { return is_tuple_; }
hlo_module_dce.cc 52 if (!xla_while->shape().IsTuple() ||
hlo_sharding_metadata.cc 188 TF_RET_CHECK(!lhs->IsTuple() && !rhs.IsTuple());
237 TF_RET_CHECK(instruction->shape().IsTuple() &&
269 if (instruction->shape().IsTuple()) {
301 if (instruction->shape().IsTuple()) {
364 TF_RET_CHECK(instruction->shape().IsTuple())
sort_simplifier.cc 34 if (!sort->shape().IsTuple()) {
112 if (new_sort->shape().IsTuple()) {
gather_expander_test.cc 92 ASSERT_TRUE(while_shape.IsTuple());
hlo_element_type_converter.cc 71 CHECK(!subshape.IsTuple());
95 CHECK(!ele_shape.IsTuple());
194 } else if (hlo->shape().IsTuple()) {
hlo_value.cc 49 instruction->shape().IsTuple() ? (" " + index.ToString()) : "";
59 string index_str = instruction->operand(operand_number)->shape().IsTuple()
90 string index_str = defining_instruction()->shape().IsTuple()
allocation_tracker.cc 145 if (!shaped_buffer->on_host_shape().IsTuple()) {
151 TF_RET_CHECK(shaped_buffer->on_device_shape().IsTuple());
  /external/tensorflow/tensorflow/c/eager/
c_api_debug.cc 86 if (padded_shape.IsTuple()) {
102 if (shape0.IsTuple() || shape1.IsTuple()) {
  /external/libcxx/test/std/utilities/tuple/tuple.tuple/tuple.cnstr/
PR22806_constrain_tuple_like_ctor.pass.cpp 32 struct IsTuple : std::false_type {};
35 struct IsTuple<Tuple, std::tuple<Args...>> : std::true_type {};
44 template <class Tuple, class = typename std::enable_if<IsTuple<Tuple>::value>::type>
57 template <class Tuple, class = typename std::enable_if<IsTuple<Tuple>::value>::type>
  /external/tensorflow/tensorflow/compiler/tf2xla/
shape_util.cc 30 if (shape.IsTuple()) {
89 if (shape.IsTuple()) {
134 if (input_shape.IsTuple()) {
142 if (shape.IsTuple()) {
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
gpu_transfer_manager.cc 133 CHECK(shape.IsTuple()) << ShapeUtil::HumanStringWithLayout(shape);
159 if (shape.IsTuple()) {
infeed_thunk.cc 41 TF_RET_CHECK(infeed_shape.IsTuple())
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
parallel_loop_emitter.cc 37 CHECK(!shape_.IsTuple());
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
infeed_test.cc 45 if (literal.shape().IsTuple()) {
  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
tensor_list_utils.cc 61 TF_RET_CHECK(list_shape.IsTuple());

Completed in 744 milliseconds

1 2 3 4