| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | ir_array.cc | 109 CHECK(LayoutUtil::HasLayout(shape)) in Index()
124 CHECK(LayoutUtil::HasLayout(shape)) in Index()
148 CHECK(LayoutUtil::HasLayout(shape)) in Index()
259 if (linear() != nullptr && LayoutUtil::HasLayout(input_shape) && in SourceIndexOfReshape()
260 LayoutUtil::HasLayout(output_shape) && in SourceIndexOfReshape()
291 if (linear() != nullptr && LayoutUtil::HasLayout(operand_shape) && in SourceIndexOfTranspose()
292 LayoutUtil::HasLayout(shape) && in SourceIndexOfTranspose()
303 CHECK(LayoutUtil::HasLayout(shape) && LayoutUtil::HasLayout(operand_shape)); in SourceIndexOfBitcast()
346 if (linear_ == nullptr || !LayoutUtil::HasLayout(operand_shape) || in SourceIndexOfBroadcast()
347 !LayoutUtil::HasLayout(shape)) { in SourceIndexOfBroadcast()
|
| /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 | 44 if (!LayoutUtil::HasLayout(shape)) { in Run()
|
| D | transfer_manager.cc | 390 if (!LayoutUtil::HasLayout(on_host_shape)) { in AllocateScopedShapedBuffer()
396 TF_RET_CHECK(LayoutUtil::HasLayout(on_device_shape)); in AllocateScopedShapedBuffer()
|
| D | layout_assignment.cc | 416 TF_RET_CHECK(LayoutUtil::HasLayout(data_shape)); in BuildHostChannelConstraints()
801 TF_RET_CHECK(LayoutUtil::HasLayout(shape_with_layout)); in CreateCopyWithNewLayout()
864 TF_RET_CHECK(LayoutUtil::HasLayout(operand->shape())); in CopyOperandIfLayoutsDiffer()
937 TF_RET_CHECK(LayoutUtil::HasLayout(instruction->shape())); in CheckLayouts()
1801 TF_RET_CHECK(LayoutUtil::HasLayout(instruction->shape())); in AssignLayouts()
|
| D | layout_assignment_test.cc | 206 EXPECT_FALSE(LayoutUtil::HasLayout( in TEST_F()
242 EXPECT_TRUE(LayoutUtil::HasLayout(tuple->shape())); in TEST_F()
|
| D | pattern_matcher.h | 849 return LayoutUtil::HasLayout(*shape) &&
854 if (!LayoutUtil::HasLayout(*shape)) {
|
| D | hlo_verifier.cc | 800 TF_RET_CHECK(LayoutUtil::HasLayout(custom_call->shape())); in HandleCustomCall()
810 TF_RET_CHECK(LayoutUtil::HasLayout(operand_shape_with_layout)); in HandleCustomCall()
|
| D | hlo_cost_analysis.cc | 139 if (!LayoutUtil::HasLayout(shape)) { in GetShapeSize()
|
| D | service.cc | 924 if (!LayoutUtil::HasLayout(return_shape)) { in TransferToClient()
|
| D | algebraic_simplifier.cc | 3881 if (!LayoutUtil::HasLayout(reshaped_shape)) { in HandleReshape()
5420 TF_RET_CHECK(LayoutUtil::HasLayout(input_shape)); in SimplifyConvToDot()
5421 TF_RET_CHECK(LayoutUtil::HasLayout(filter_shape)); in SimplifyConvToDot()
5422 TF_RET_CHECK(LayoutUtil::HasLayout(convolution_shape)); in SimplifyConvToDot()
|
| D | hlo_parser.cc | 2288 if (!LayoutUtil::HasLayout(shape)) { in ParseInstructionRhs()
2301 if (!LayoutUtil::HasLayout(operand_shape_with_layout)) { in ParseInstructionRhs()
|
| D | hlo_evaluator.cc | 1948 if (!LayoutUtil::HasLayout(instruction->shape())) { in HandleFusion()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | layout_util.h | 114 static bool HasLayout(const Shape& shape);
117 static bool HasLayout(const ProgramShape& program_shape);
|
| D | layout_util.cc | 295 /* static */ bool LayoutUtil::HasLayout(const Shape& shape) { in HasLayout() function in xla::LayoutUtil
299 [](const Shape& s) { return HasLayout(s); }); in HasLayout()
307 /* static */ bool LayoutUtil::HasLayout(const ProgramShape& program_shape) { in HasLayout() function in xla::LayoutUtil
309 if (!LayoutUtil::HasLayout(parameter_shape)) { in HasLayout()
313 return LayoutUtil::HasLayout(program_shape.result()); in HasLayout()
|
| D | layout_util_test.cc | 203 EXPECT_TRUE(LayoutUtil::HasLayout(shape)); in TEST_F()
209 EXPECT_FALSE(LayoutUtil::HasLayout(shape)); in TEST_F()
218 EXPECT_TRUE(LayoutUtil::HasLayout(shape)); in TEST_F()
220 EXPECT_TRUE(LayoutUtil::HasLayout(shape)); in TEST_F()
|
| D | literal.cc | 177 CHECK(LayoutUtil::HasLayout(*shape_)); in Literal()
360 if (!LayoutUtil::HasLayout(shape)) { in CreateFromProto()
1277 CHECK(LayoutUtil::HasLayout(literal.shape())); in ToStringHelper()
1278 CHECK(LayoutUtil::HasLayout(subshape)); in ToStringHelper()
1295 CHECK(LayoutUtil::HasLayout(this->shape())); in ToString()
1307 CHECK(LayoutUtil::HasLayout(this->shape())); in ToStringWithoutShape()
1319 CHECK(LayoutUtil::HasLayout(this->shape())); in ToStringWithLayout()
2136 TF_RET_CHECK(LayoutUtil::HasLayout(shape)); in CopyFromProto()
2304 CHECK(LayoutUtil::HasLayout(*shape_)); in MutableBorrowingLiteral()
2315 CHECK(LayoutUtil::HasLayout(*shape_)); in operator =()
[all …]
|
| D | shape_layout.cc | 66 bool ShapeLayout::LayoutIsSet() const { return LayoutUtil::HasLayout(shape_); } in LayoutIsSet()
|
| D | shape.cc | 103 if (LayoutUtil::HasLayout(*this)) { in DeleteDimension()
|
| D | shape_util.cc | 622 } else if (shape.IsArray() && LayoutUtil::HasLayout(shape)) { in HumanStringWithLayout()
1229 CHECK(LayoutUtil::HasLayout(input_shape) && in TransposeIsBitcast()
1230 LayoutUtil::HasLayout(output_shape)); in TransposeIsBitcast()
1260 CHECK(LayoutUtil::HasLayout(input_shape)); in ReshapeIsBitcast()
1261 CHECK(LayoutUtil::HasLayout(output_shape)); in ReshapeIsBitcast()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emission_utils.cc | 30 CHECK(!LayoutUtil::HasLayout(shape) || LayoutUtil::IsDense(shape.layout())); in GetMinimumAlignmentForArray()
|
| /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 | 1666 if (!LayoutUtil::HasLayout(shape)) { in Infeed()
1739 if (!LayoutUtil::HasLayout(shape)) { in InfeedWithToken()
1776 if (!LayoutUtil::HasLayout(shape_with_layout)) { in Outfeed()
1825 if (!LayoutUtil::HasLayout(shape_with_layout)) { in OutfeedWithToken()
1895 if (!LayoutUtil::HasLayout(shape)) { in CustomCall()
1908 if (!LayoutUtil::HasLayout(operand_shape)) { in CustomCall()
1981 if (!LayoutUtil::HasLayout(shape)) { in CustomCall()
1995 if (!LayoutUtil::HasLayout(operand_shape)) { in CustomCall()
2812 if (!LayoutUtil::HasLayout(*shape_with_layout)) { in AllReduce()
3242 if (!LayoutUtil::HasLayout(shape_with_layout)) { in SendToHost()
[all …]
|