| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_layout_assignment_test.cc | 159 if (input_shape.has_layout()) { in TEST_F()
164 if (result_shape.has_layout()) { in TEST_F()
228 if (input_shape.has_layout()) { in TEST_F()
233 if (result_shape.has_layout()) { in TEST_F()
310 if (input_shape.has_layout()) { in TEST_F()
315 if (result_shape.has_layout()) { in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | layout_util.cc | 173 if (shape.has_layout()) { in ValidateLayoutInShape()
182 if (!shape.has_layout()) { in ValidateLayoutInShape()
192 if (shape.has_layout()) { in ValidateLayoutInShape()
270 return shape.IsArray() && shape.has_layout() && IsDense(shape.layout()); in IsDenseArray()
298 return shape.has_layout() && shape.layout().format() != INVALID_FORMAT; in HasLayout()
375 if (src.has_layout()) { in CopyLayoutInternal()
|
| D | layout_util_test.cc | 81 EXPECT_TRUE(dst.has_layout()); in TEST_F()
84 EXPECT_FALSE(dst.has_layout()); in TEST_F()
204 EXPECT_TRUE(shape.tuple_shapes(0).has_layout()); in TEST_F()
205 EXPECT_TRUE(shape.tuple_shapes(2).tuple_shapes(1).has_layout()); in TEST_F()
210 EXPECT_FALSE(shape.tuple_shapes(0).has_layout()); in TEST_F()
211 EXPECT_FALSE(shape.tuple_shapes(2).tuple_shapes(1).has_layout()); in TEST_F()
|
| D | shape.cc | 53 if (shape_proto.has_layout()) { in Shape()
72 if (has_layout()) { in ToProto()
|
| D | shape.h | 129 bool has_layout() const { return layout_.format() != INVALID_FORMAT; } in has_layout() function
|
| D | shape_util.cc | 112 if (!shape.has_layout()) { in MakeShapeWithLayoutInternal()
691 if (shape.has_layout()) { in ValidateShapeWithOptionalLayoutInternal()
969 if (shape.has_layout()) { in PermuteDimensions()
|
| D | shape_util_test.cc | 298 ASSERT_TRUE(scalar_default_layout.has_layout()) in TEST()
303 ASSERT_TRUE(scalar_empty_min2maj.has_layout()) in TEST()
|
| /external/tensorflow/tensorflow/core/profiler/convert/ |
| D | op_metrics_db_combiner.cc | 32 if (!dst->has_layout() && src.has_layout()) { in CombineOpMetrics()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | shape_util.cc | 76 TF_ASSIGN_OR_RETURN(bool has_layout, MakeLayout(minor_to_major, &layout)); in AssignLayout()
77 if (!has_layout && layout_func) { in AssignLayout()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_matchers.cc | 260 if (!copy_done->shape().has_layout()) { in MatchAndExplain()
274 if (!copy_start_operand->shape().has_layout()) { in MatchAndExplain()
|
| D | layout_assignment.cc | 322 if (subshape.IsArray() && subshape.has_layout()) { in SetInstructionLayout()
770 !subshape.has_layout()) { in CheckParameterLayout()
1611 TF_RET_CHECK(!subshape.has_layout()); in InferArrayLayout()
1626 if (!source_buffer->shape().has_layout()) { in InferArrayLayout()
1741 if (subshape->has_layout() || !subshape->IsArray()) { in AssignLayouts()
2020 if (!subshape.has_layout()) { in PropagateComputationLayouts()
|
| D | while_util.cc | 229 if (!shape.has_layout()) { in MakeLoopStateShapeWithLayout()
|
| D | memory_space_assignment.cc | 539 if (subshape.has_layout() && subshape.layout().memory_space() == in AddInputAndOutputRequiredAssignments()
563 if (subshape.has_layout() && subshape.layout().memory_space() == in AddInputAndOutputRequiredAssignments()
585 return shape.has_layout() && in AreIntervalsReservedInAlternateMemory()
|
| D | buffer_assignment.h | 570 if (defining_position.shape().has_layout()) { in DefaultColorer()
|
| D | pattern_matcher.h | 965 << (shape->has_layout() ? ShapeUtil::HumanStringWithLayout(*shape)
980 << (shape->has_layout() ? ShapeUtil::HumanStringWithLayout(*shape)
|
| D | buffer_assignment.cc | 260 if (shape->has_layout()) { in AddAssignment()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_utils.cc | 46 if (!shape.has_layout() || shape.layout().minor_to_major().empty()) { in GetPermutationIfAvailable()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 804 CHECK(!lhs_shape.has_layout() || IsSimpleLayout(lhs_shape.layout())) in AreGemmShapes()
806 CHECK(!rhs_shape.has_layout() || IsSimpleLayout(rhs_shape.layout())) in AreGemmShapes()
808 CHECK(!output_shape.has_layout() || IsSimpleLayout(output_shape.layout())) in AreGemmShapes()
938 CHECK(shape.has_layout() && in CollapseFirstNDims()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | dlpack.cc | 154 CHECK(shape.has_layout()); in StridesForShape()
|
| D | types.cc | 186 CHECK(shape.has_layout()); in ByteStridesForShape()
|
| D | local_client.cc | 1067 if (subshape->IsArray() && !subshape->has_layout()) { in Compile()
|
| D | xla.cc | 418 if (!subshape->has_layout()) { in PYBIND11_MODULE()
|
| /external/mesa3d/src/compiler/glsl/ |
| D | glsl_parser.yy | 853 if (block->layout.has_layout() || block->layout.has_memory()) {
2091 $$.merge_qualifier(& @1, state, $2, false, $2.has_layout());
2107 $2.has_interpolation() || $2.has_layout())) {
2130 $2.has_layout() || $2.has_auxiliary_storage())) {
2863 block->layout.has_layout())) {
|
| D | ast.h | 808 bool has_layout() const;
|
| D | ast_type.cpp | 61 ast_type_qualifier::has_layout() const in has_layout() function in ast_type_qualifier
|