| /external/ruy/example/ |
| D | example.cc | 27 ruy::MakeSimpleLayout(2, 2, ruy::Order::kRowMajor, lhs.mutable_layout()); in ExampleMulFloat()
30 ruy::MakeSimpleLayout(2, 2, ruy::Order::kColMajor, rhs.mutable_layout()); in ExampleMulFloat()
33 ruy::MakeSimpleLayout(2, 2, ruy::Order::kColMajor, dst.mutable_layout()); in ExampleMulFloat()
52 ruy::MakeSimpleLayout(2, 2, ruy::Order::kRowMajor, lhs.mutable_layout()); in ExampleMulFloatWithBiasAddAndClamp()
55 ruy::MakeSimpleLayout(2, 2, ruy::Order::kColMajor, rhs.mutable_layout()); in ExampleMulFloatWithBiasAddAndClamp()
58 ruy::MakeSimpleLayout(2, 2, ruy::Order::kColMajor, dst.mutable_layout()); in ExampleMulFloatWithBiasAddAndClamp()
79 ruy::MakeSimpleLayout(2, 2, ruy::Order::kRowMajor, lhs.mutable_layout()); in ExampleMulUint8AsymmetricQuantized()
83 ruy::MakeSimpleLayout(2, 2, ruy::Order::kColMajor, rhs.mutable_layout()); in ExampleMulUint8AsymmetricQuantized()
87 ruy::MakeSimpleLayout(2, 2, ruy::Order::kColMajor, dst.mutable_layout()); in ExampleMulUint8AsymmetricQuantized()
110 ruy::MakeSimpleLayout(2, 2, ruy::Order::kRowMajor, lhs.mutable_layout()); in ExampleMulInt8PerChannelQuantized()
[all …]
|
| D | parametrized_example.cc | 94 parse_order(params.lhs_order), lhs.mutable_layout()); in run()
96 parse_order(params.rhs_order), rhs.mutable_layout()); in run()
98 parse_order(params.dst_order), dst.mutable_layout()); in run()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | layout_util_test.cc | 33 *shape.mutable_layout() = LayoutUtil::MakeLayout(minor_to_major); in MakeShapeWithLayout()
145 *src.mutable_layout() = LayoutUtil::MakeLayout({1, 2, 3, 4}); in TEST_F()
280 tile = shape.mutable_layout()->add_tiles(); in TEST_F()
287 shape.mutable_layout()->clear_tiles(); in TEST_F()
288 tile = shape.mutable_layout()->add_tiles(); in TEST_F()
295 tile = shape.mutable_layout()->add_tiles(); in TEST_F()
298 tile = shape.mutable_layout()->add_tiles(); in TEST_F()
306 tile = shape.mutable_layout()->add_tiles(); in TEST_F()
313 shape.mutable_layout()->clear_tiles(); in TEST_F()
314 tile = shape.mutable_layout()->add_tiles(); in TEST_F()
[all …]
|
| D | shape_layout.cc | 95 *shape_.mutable_layout() = layout; in ResetLayout()
102 *ShapeUtil::GetMutableSubshape(&shape_, shape_index)->mutable_layout() = in ResetLayout()
|
| D | index_util_test.cc | 28 shape->mutable_layout()->clear_minor_to_major(); in SetMinorToMajorLayout()
30 shape->mutable_layout()->add_minor_to_major(dimension); in SetMinorToMajorLayout()
|
| D | shape_util.cc | 141 *shape.mutable_layout() = LayoutUtil::MakeLayout( in MakeShapeWithLayoutInternal()
225 auto layout = shape->mutable_layout(); in FillNewShape()
336 *ret.mutable_layout() = LayoutUtil::MoveDimToMajor(ret.layout(), dim); in MoveDimToMajor()
344 *ret.mutable_layout() = LayoutUtil::MakeLayout(minor_to_major); in MoveDimToMajor()
365 new_shape.mutable_layout()->mutable_tiles()->assign( in MakeShapeWithDescendingLayoutAndSamePhysicalLayout()
367 new_shape.mutable_layout()->set_element_size_in_bits( in MakeShapeWithDescendingLayoutAndSamePhysicalLayout()
446 shape->mutable_layout()->add_minor_to_major(shape->rank()); in AppendMajorDimension()
1140 Layout* new_layout = new_shape.mutable_layout(); in PermuteDimensions()
1496 *output_shape_with_layout.mutable_layout() = Layout{layout}; in AlignLayouts()
1768 subshape->mutable_layout()->clear_tiles(); in DeviceShapeToHostShape()
[all …]
|
| D | shape.cc | 54 *mutable_layout() = Layout::CreateFromProto(shape_proto.layout()); in Shape()
73 *proto.mutable_layout() = layout().ToProto(); in ToProto()
|
| D | shape_test.cc | 233 *shape.mutable_layout() = Layout({2, 1, 0}); in BM_ShapeCopy()
239 *shape.mutable_layout() = Layout({2, 1, 0}, {Tile({2, 128})}); in BM_ShapeCopy()
|
| D | layout_util.cc | 152 shape->mutable_layout()->set_format(DENSE); in SetToDefaultLayout()
153 auto* minor_to_major = shape->mutable_layout()->mutable_minor_to_major(); in SetToDefaultLayout()
388 *dst->mutable_layout() = src.layout(); in CopyLayoutInternal()
|
| D | packed_literal_reader.cc | 50 *literal_shape.mutable_layout() = *layout; in Read()
|
| D | shape_util_test.cc | 131 auto layout_1 = shape_1.mutable_layout(); in TEST()
137 auto layout_2 = shape_2.mutable_layout(); in TEST()
214 *shape_a.mutable_layout() = Layout({1, 0}); in TEST()
216 *shape_b.mutable_layout() = Layout({0, 1}); in TEST()
218 *shape_c.mutable_layout() = Layout({0, 1}); in TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_layout_assignment.cc | 178 std::tie(*input_shape->mutable_layout(), in AddBackendConstraintsToDnnConvCustomCall()
179 *filter_shape->mutable_layout(), in AddBackendConstraintsToDnnConvCustomCall()
180 *output_shape->mutable_layout()), in AddBackendConstraintsToDnnConvCustomCall()
277 *shape.mutable_layout() = keys_layout; in AddBackendConstraints()
304 int n = shape->mutable_layout()->minor_to_major_size(); in AddBackendConstraints()
306 std::swap(shape->mutable_layout()->mutable_minor_to_major()->at(0), in AddBackendConstraints()
307 shape->mutable_layout()->mutable_minor_to_major()->at(1)); in AddBackendConstraints()
409 *shape_with_layout.mutable_layout() = buffer_constraint.layout(); in PropagateBufferConstraint()
|
| D | cusolver_rewriter.cc | 43 int n = shape->mutable_layout()->minor_to_major_size(); in SetFortranLayout()
45 std::swap(shape->mutable_layout()->mutable_minor_to_major()->at(0), in SetFortranLayout()
46 shape->mutable_layout()->mutable_minor_to_major()->at(1)); in SetFortranLayout()
|
| D | gpu_layout_assignment_test.cc | 47 *ashape_in_row_major.mutable_layout() = LayoutUtil::MakeLayout({1, 0}); in TEST_F()
48 *ashape_in_col_major.mutable_layout() = LayoutUtil::MakeLayout({0, 1}); in TEST_F()
104 *shapes.back().mutable_layout() = LayoutUtil::MakeLayout(layout_vec); in AllLayoutsOf()
|
| D | elemental_ir_emitter.cc | 173 *shape.mutable_layout() = in GetSourceIndexOfBitcast()
175 *operand_shape.mutable_layout() = in GetSourceIndexOfBitcast()
|
| /external/ruy/ruy/ |
| D | test_overflow_dst_zero_point.cc | 74 ruy::MakeSimpleLayout(1, 1, ruy::Order::kColMajor, lhs.mutable_layout()); in TestOverflowingAdditionOfDestinationZeroPoint()
78 ruy::MakeSimpleLayout(1, cols, ruy::Order::kColMajor, rhs.mutable_layout()); in TestOverflowingAdditionOfDestinationZeroPoint()
82 ruy::MakeSimpleLayout(1, cols, ruy::Order::kColMajor, dst.mutable_layout()); in TestOverflowingAdditionOfDestinationZeroPoint()
|
| D | prepacked_cache_test.cc | 240 ruy::MakeSimpleLayout(2, 2, ruy::Order::kRowMajor, lhs.mutable_layout()); in TestCachePolicies()
243 ruy::MakeSimpleLayout(2, 1, ruy::Order::kColMajor, rhs.mutable_layout()); in TestCachePolicies()
246 ruy::MakeSimpleLayout(2, 1, ruy::Order::kColMajor, dst.mutable_layout()); in TestCachePolicies()
281 ruy::MakeSimpleLayout(2, 2, ruy::Order::kRowMajor, lhs.mutable_layout()); in TEST()
284 ruy::MakeSimpleLayout(2, 1, ruy::Order::kColMajor, rhs.mutable_layout()); in TEST()
287 ruy::MakeSimpleLayout(2, 1, ruy::Order::kColMajor, dst.mutable_layout()); in TEST()
|
| D | perchannel_buffers_reallocation_test.cc | 54 lhs.mutable_layout()); in TestPerChannelBuffersReallocation()
59 rhs.mutable_layout()); in TestPerChannelBuffersReallocation()
65 dst.mutable_layout()); in TestPerChannelBuffersReallocation()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment_test.cc | 108 *shape.mutable_layout() = layout; in TEST_F()
138 *col_major_shape.mutable_layout() = col_major_layout; in TEST_F()
143 *row_major_shape.mutable_layout() = row_major_layout; in TEST_F()
190 *shape.mutable_layout() = layout; in TEST_F()
368 *ashape_with_layout.mutable_layout() = LayoutUtil::MakeLayout({0, 2, 1, 3}); in TEST_F()
369 *bshape_with_layout.mutable_layout() = LayoutUtil::MakeLayout({2, 1, 0}); in TEST_F()
409 *ashape_with_layout.mutable_layout() = LayoutUtil::MakeLayout({1, 0}); in TEST_F()
410 *bshape_with_layout.mutable_layout() = LayoutUtil::MakeLayout({0, 1}); in TEST_F()
446 *input_shape_with_layout.mutable_layout() = LayoutUtil::MakeLayout({1, 0}); in TEST_F()
447 *output_shape_with_layout.mutable_layout() = in TEST_F()
[all …]
|
| D | result_caster.cc | 58 *inferred_shape.mutable_layout() = instruction->shape().layout(); in ExpandInstruction()
|
| D | hlo_cse_test.cc | 433 *exp1->mutable_shape()->mutable_layout() = LayoutUtil::MakeLayout({0, 1}); in TEST_F()
437 *exp2->mutable_shape()->mutable_layout() = LayoutUtil::MakeLayout({1, 0}); in TEST_F()
464 *exp1->mutable_shape()->mutable_layout() = LayoutUtil::MakeLayout({0, 1}); in TEST_F()
468 *exp2->mutable_shape()->mutable_layout() = LayoutUtil::MakeLayout({1, 0}); in TEST_F()
|
| D | memory_space_propagation.cc | 75 shape->mutable_layout()->set_memory_space(memory_space); in Propagate()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | reduce_hlo_test.cc | 85 *reduce_output_shape->mutable_layout() = in XLA_TEST_P()
90 *reduce_input_shape->mutable_layout() = in XLA_TEST_P()
|
| D | test_utils.cc | 286 new_shape.mutable_layout()->clear_tiles(); in MakeFakeLiteralInternal()
287 new_shape.mutable_layout()->set_element_size_in_bits(0); in MakeFakeLiteralInternal()
388 new_shape.mutable_layout()->clear_tiles(); in MakeFakeLiteralInternalWithBounds()
389 new_shape.mutable_layout()->set_element_size_in_bits(0); in MakeFakeLiteralInternalWithBounds()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_layout_assignment.cc | 84 *new_shape.mutable_layout() = LayoutUtil::MakeLayout(dimension_order); in RowMajorShape()
92 *new_shape.mutable_layout() = LayoutUtil::MakeLayout(dimension_order); in ColMajorShape()
|