Home
last modified time | relevance | path

Searched refs:IsToken (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dinfeed_thunk.cc45 TF_RET_CHECK(infeed_shape.tuple_shapes(1).IsToken()) in ExecuteOnStream()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_alias_analysis.cc534 if (buffer.values().front()->shape().IsToken()) { in HasLiveRangeInterference()
538 DCHECK(value->shape().IsToken()); in HasLiveRangeInterference()
Dcopy_insertion.cc489 if (value_a->shape().IsToken()) { in CopyRemover()
Dhlo_computation.cc646 if (instruction->shape().IsToken()) { in DeepCopyHelper()
Dhlo_verifier.cc1057 if (subshape.IsToken()) { in ShapeContainsToken()
/external/tensorflow/tensorflow/compiler/xla/
Dshape.h58 bool IsToken() const { return element_type() == TOKEN; } in IsToken() function
Dlayout_util.cc116 if (shape.IsOpaque() || shape.IsToken()) { in GetDefaultLayoutForShape()
Dliteral.cc1177 } else if (subshape.IsToken()) { in ToStringHelper()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dif_op.cc251 OP_REQUIRES(ctx, shape_or.ValueOrDie().IsToken(), in Compile()
Dcase_op.cc263 OP_REQUIRES(ctx, shape_or.ValueOrDie().IsToken(), in Compile()
Dwhile_op.cc545 OP_REQUIRES(ctx, shape_or.ValueOrDie().IsToken(), in Compile()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler_test.cc1493 EXPECT_TRUE(result.xla_input_shapes[1].IsToken()); in TEST_F()
1497 .IsToken()); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc1481 if (!operand_shape.IsToken()) { in AfterAll()