| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | ir_array.cc | 114 CHECK(LayoutUtil::HasLayout(shape)) in Index()
135 CHECK(LayoutUtil::HasLayout(shape)) in Index()
155 CHECK(LayoutUtil::HasLayout(shape)) in Index()
179 CHECK(LayoutUtil::HasLayout(shape)) in Index()
288 if (linear() != nullptr && LayoutUtil::HasLayout(input_shape) && in SourceIndexOfReshape()
289 LayoutUtil::HasLayout(output_shape) && in SourceIndexOfReshape()
320 if (linear() != nullptr && LayoutUtil::HasLayout(operand_shape) && in SourceIndexOfTranspose()
321 LayoutUtil::HasLayout(shape) && in SourceIndexOfTranspose()
332 CHECK(LayoutUtil::HasLayout(shape) && LayoutUtil::HasLayout(operand_shape)); in SourceIndexOfBitcast()
375 if (linear_ == nullptr || !LayoutUtil::HasLayout(operand_shape) || in SourceIndexOfBroadcast()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | optimize_input_output_buffer_alias.cc | 44 TF_RET_CHECK(LayoutUtil::HasLayout(input_shape)); in Build()
45 TF_RET_CHECK(LayoutUtil::HasLayout(output_shape)); in Build()
|
| D | zero_sized_hlo_elimination.cc | 47 if (!LayoutUtil::HasLayout(shape)) { in Run()
|
| D | transfer_manager.cc | 384 if (!LayoutUtil::HasLayout(on_host_shape)) { in AllocateScopedShapedBuffer()
392 TF_RET_CHECK(LayoutUtil::HasLayout(on_device_shape)); in AllocateScopedShapedBuffer()
|
| D | layout_assignment.cc | 518 TF_RET_CHECK(LayoutUtil::HasLayout(data_shape)); in BuildHostChannelConstraints()
991 TF_RET_CHECK(LayoutUtil::HasLayout(shape_with_layout)); in CreateCopyWithNewLayout()
1054 TF_RET_CHECK(LayoutUtil::HasLayout(operand->shape())); in CopyOperandIfLayoutsDiffer()
1139 TF_RET_CHECK(LayoutUtil::HasLayout(instruction->shape())); in CheckLayouts()
2036 TF_RET_CHECK(LayoutUtil::HasLayout(instruction->shape())); in AssignLayouts()
|
| D | layout_assignment_test.cc | 207 EXPECT_FALSE(LayoutUtil::HasLayout( in TEST_F()
243 EXPECT_TRUE(LayoutUtil::HasLayout(tuple->shape())); in TEST_F()
|
| D | pattern_matcher.h | 903 return LayoutUtil::HasLayout(*shape) &&
908 if (!LayoutUtil::HasLayout(*shape)) {
|
| D | hlo_cost_analysis.cc | 139 if (!LayoutUtil::HasLayout(shape)) { in GetShapeSize()
|
| D | hlo_verifier.cc | 1170 TF_RET_CHECK(LayoutUtil::HasLayout(custom_call->shape())); in HandleCustomCall()
1180 TF_RET_CHECK(LayoutUtil::HasLayout(operand_shape_with_layout)); in HandleCustomCall()
|
| D | service.cc | 987 if (!LayoutUtil::HasLayout(return_shape)) { in TransferToClient()
|
| D | algebraic_simplifier.cc | 4257 if (!LayoutUtil::HasLayout(reshaped_shape)) { in HandleReshape()
6563 TF_RET_CHECK(LayoutUtil::HasLayout(input_shape)); in SimplifyConvToDot()
6564 TF_RET_CHECK(LayoutUtil::HasLayout(filter_shape)); in SimplifyConvToDot()
6565 TF_RET_CHECK(LayoutUtil::HasLayout(convolution_shape)); in SimplifyConvToDot()
|
| D | hlo_parser.cc | 2651 if (!LayoutUtil::HasLayout(*shape)) { in CreateInstruction()
2664 if (!LayoutUtil::HasLayout(operand_shape_with_layout)) { in CreateInstruction()
|
| D | hlo_evaluator.cc | 3153 if (!LayoutUtil::HasLayout(instruction->shape())) { in HandleFusion()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | layout_util.h | 149 static bool HasLayout(const Shape& shape);
152 static bool HasLayout(const ProgramShape& program_shape);
|
| D | layout_util.cc | 359 /* static */ bool LayoutUtil::HasLayout(const Shape& shape) { in HasLayout() function in xla::LayoutUtil
363 [](const Shape& s) { return HasLayout(s); }); in HasLayout()
371 /* static */ bool LayoutUtil::HasLayout(const ProgramShape& program_shape) { in HasLayout() function in xla::LayoutUtil
373 if (!LayoutUtil::HasLayout(parameter_shape)) { in HasLayout()
377 return LayoutUtil::HasLayout(program_shape.result()); in HasLayout()
|
| D | layout_util_test.cc | 245 EXPECT_TRUE(LayoutUtil::HasLayout(shape)); in TEST_F()
251 EXPECT_FALSE(LayoutUtil::HasLayout(shape)); in TEST_F()
260 EXPECT_TRUE(LayoutUtil::HasLayout(shape)); in TEST_F()
262 EXPECT_TRUE(LayoutUtil::HasLayout(shape)); in TEST_F()
|
| D | shape_layout.cc | 90 bool ShapeLayout::LayoutIsSet() const { return LayoutUtil::HasLayout(shape_); } in LayoutIsSet()
|
| D | literal.cc | 281 LayoutUtil::HasLayout(*shape_)); in Literal()
466 if (!LayoutUtil::HasLayout(shape)) { in CreateFromProto()
1414 CHECK(LayoutUtil::HasLayout(literal.shape())); in ToStringHelper()
1415 CHECK(LayoutUtil::HasLayout(subshape)); in ToStringHelper()
1442 CHECK(LayoutUtil::HasLayout(this->shape())); in ToString()
1454 CHECK(LayoutUtil::HasLayout(this->shape())); in ToStringWithoutShape()
1466 CHECK(LayoutUtil::HasLayout(this->shape())); in ToStringWithLayout()
2386 TF_RET_CHECK(LayoutUtil::HasLayout(shape)); in CopyFromProto()
2590 CHECK(LayoutUtil::HasLayout(*shape_)); in MutableBorrowingLiteral()
2601 CHECK(LayoutUtil::HasLayout(*shape_)); in operator =()
[all …]
|
| D | shape.cc | 128 if (LayoutUtil::HasLayout(*this)) { in DeleteDimension()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | shape_util.cc | 37 } else if (xla::LayoutUtil::HasLayout(shape)) { in PopulateInfeedLayoutVector()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1751 if (!LayoutUtil::HasLayout(shape)) { in Infeed()
1824 if (!LayoutUtil::HasLayout(shape)) { in InfeedWithToken()
1862 if (!LayoutUtil::HasLayout(shape_with_layout)) { in Outfeed()
1911 if (!LayoutUtil::HasLayout(shape_with_layout)) { in OutfeedWithToken()
1983 if (!LayoutUtil::HasLayout(shape)) { in CustomCall()
1996 if (!LayoutUtil::HasLayout(operand_shape)) { in CustomCall()
2091 if (!LayoutUtil::HasLayout(shape)) { in CustomCall()
2104 if (!LayoutUtil::HasLayout(operand_shape)) { in CustomCall()
2968 if (!LayoutUtil::HasLayout(*shape_with_layout)) { in AllReduce()
3472 if (!LayoutUtil::HasLayout(shape_with_layout)) { in SendToHost()
[all …]
|