| /external/tensorflow/tensorflow/compiler/xla/service/ |
| computation_layout.h | 65 const Shape& result_shape() const { return result_layout_.shape(); } function in class:xla::ComputationLayout
|
| executable.h | 151 const Shape& result_shape() const { function in class:xla::Executable
152 return hlo_module_->config().entry_computation_layout().result_shape();
|
| hlo_module.cc | 297 const Shape& result_shape = local
299 TF_RET_CHECK(ShapeUtil::Equal(expected_program_shape.result(), result_shape))
303 << ", actual: " << ShapeUtil::HumanStringWithLayout(result_shape);
|
| layout_assignment_test.cc | 245 Shape result_shape = local
248 result_shape));
252 EXPECT_TRUE(LayoutUtil::LayoutsInShapesEqual(result_shape, select->shape()));
280 Shape result_shape = nested_tuple->shape(); local
281 *ShapeUtil::GetMutableSubshape(&result_shape, /*index=*/{0, 0}) =
283 *ShapeUtil::GetMutableSubshape(&result_shape, /*index=*/{1, 0}) =
286 result_shape));
309 EXPECT_TRUE(ShapeUtil::Equal(result_shape, root->shape()));
310 EXPECT_TRUE(ShapeUtil::Equal(ShapeUtil::GetSubshape(result_shape, {0}),
312 EXPECT_TRUE(ShapeUtil::Equal(ShapeUtil::GetSubshape(result_shape, {1})
[all...] |
| layout_assignment.cc | 558 DCHECK(ShapeUtil::Compatible(body_layout.result_shape(),
560 DCHECK(ShapeUtil::Compatible(body_layout.result_shape(),
562 DCHECK(ShapeUtil::Compatible(body_layout.result_shape(), init->shape()));
566 if (!ShapeUtil::Equal(body_layout.result_shape(),
572 ShapeUtil::HumanString(body_layout.result_shape()).c_str(),
583 ShapeUtil::HumanString(body_layout.result_shape()).c_str());
589 body_layout.result_shape(), instruction));
591 body_layout.result_shape(), instruction, 0));
615 true_computation_layout.result_shape(), instruction));
628 Shape result_shape = ShapeUtil::MakeShapeWithDescendingLayout local
[all...] |
| shape_inference_test.cc | 1224 Shape result_shape = ShapeUtil::MakeTupleShape({s32_, vector_32_}); local
1236 Shape result_shape = ShapeUtil::MakeTupleShape({s32_, vector_32_}); local
[all...] |
| algebraic_simplifier_test.cc | 1091 Shape result_shape = ShapeUtil::MakeShape(F32, {3 * kParamLength}); local
1121 Shape result_shape = ShapeUtil::MakeShape(F32, {0}); local
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| reshape_util.cc | 139 Tensor *result_shape = nullptr; local
142 &result_shape));
143 auto output_shape_vec = result_shape->vec<int64>();
|
| gather_op.cc | 89 TensorShape result_shape; variable
93 result_shape.AddDim(params.dim_size(i));
96 result_shape.AppendShape(indices.shape());
98 result_shape.AddDim(params.dim_size(i));
103 OP_REQUIRES_OK(c, c->allocate_output(0, result_shape, &out));
|
| dynamic_stitch_op.cc | 127 TensorShape result_shape; local
128 result_shape.AddDim(*first_dim_size);
130 result_shape.AddDim(data0.dim_size(d));
132 OP_REQUIRES_OK(c, c->allocate_output(0, result_shape, result_ptr));
|
| resource_variable_ops.cc | 506 TensorShape result_shape = indices.shape(); variable
508 result_shape.AddDim(params.dim_size(i));
512 OP_REQUIRES_OK(c, c->allocate_output(0, result_shape, &out));
|
| /external/tensorflow/tensorflow/core/util/ |
| tensor_slice_set.cc | 35 TensorShape result_shape; local
36 TF_RETURN_IF_ERROR(slice.SliceTensorShape(shape_, &result_shape));
57 result_shape.num_elements()};
|
| bcast.h | 99 const Vec& result_shape() const { return result_; } function in class:tensorflow::BCast
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_layout_assignment_test.cc | 61 Shape result_shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 24}, {0, 1}); local
67 HloInstruction::CreateCanonicalDot(result_shape, dot_lhs, dot_rhs));
76 ShapeLayout(LayoutUtil::GetWithDefaultLayout(result_shape));
96 Shape result_shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 24}, {0, 1}); local
104 HloInstruction::CreateCanonicalDot(result_shape, dot_a_lhs, dot_rhs));
106 HloInstruction::CreateCanonicalDot(result_shape, dot_b_lhs, dot_rhs));
108 result_shape, HloOpcode::kAdd, dot_a_result, dot_b_result));
117 ShapeLayout(LayoutUtil::GetWithDefaultLayout(result_shape));
180 Shape result_shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 24}, {0, 1}); local
186 HloInstruction::CreateCanonicalDot(result_shape, dot_lhs, dot_rhs))
219 Shape result_shape = ShapeUtil::MakeShape(F32, {1, 24}); local
[all...] |
| cpu_instruction_fusion_test.cc | 250 Shape result_shape = ShapeUtil::MakeShape(F32, {4}); local
258 HloInstruction::CreateUnary(result_shape, HloOpcode::kBitcast, exp1));
260 HloInstruction::CreateUnary(result_shape, HloOpcode::kNegate, bitcast2));
303 Shape result_shape = ShapeUtil::MakeShape(F32, {8, 8}); local
307 HloInstruction::CreateBroadcast(result_shape, param0, {1}));
309 result_shape, HloOpcode::kNegate, broadcast1));
323 Shape result_shape = ShapeUtil::MakeShape(F32, {2}); local
329 HloInstruction::CreateDynamicSlice(result_shape, param0, param1, {2}));
331 result_shape, HloOpcode::kNegate, dynamic_slice2));
362 Shape result_shape = ShapeUtil::MakeShape(F32, {16}) local
418 Shape result_shape = ShapeUtil::MakeShape(F32, {4, 3}); local
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| reduce_window_test.cc | 330 Shape result_shape = local
332 std::unique_ptr<Literal> expected = Literal::CreateFromShape(result_shape);
[all...] |
| while_test.cc | 54 auto result_shape = ShapeUtil::MakeShape(S32, {}); local
60 auto prev = builder.Parameter(0, result_shape, "prev");
69 auto prev = builder.Parameter(0, result_shape, "prev");
91 auto result_shape = ShapeUtil::MakeShape(S64, {}); local
97 auto prev = builder.Parameter(0, result_shape, "prev");
106 auto prev = builder.Parameter(0, result_shape, "prev");
122 auto result_shape = ShapeUtil::MakeShape(S32, {}); local
129 auto prev = builder.Parameter(0, result_shape, "prev");
138 auto prev = builder.Parameter(0, result_shape, "prev");
156 auto result_shape = ShapeUtil::MakeShape(PRED, {}) local
194 Shape result_shape = ShapeUtil::MakeShape(F32, {0}); local
247 Shape result_shape = ShapeUtil::MakeShape(F32, {8}); local
306 Shape result_shape = ShapeUtil::MakeShape(F32, {8}); local
364 Shape result_shape = ShapeUtil::MakeTupleShape(shape_elements); local
417 Shape result_shape = ShapeUtil::MakeTupleShape(shape_elements); local
473 Shape result_shape = ShapeUtil::MakeTupleShape(shape_elements); local
522 Shape result_shape = ShapeUtil::MakeTupleShape(shape_elements); local
569 Shape result_shape = ShapeUtil::MakeTupleShape(shape_elements); local
630 Shape result_shape = ShapeUtil::MakeTupleShape(shape_elements); local
709 Shape result_shape = ShapeUtil::MakeTupleShape(shape_elements); local
776 Shape result_shape = ShapeUtil::MakeTupleShape(shape_elements); local
843 Shape result_shape = ShapeUtil::MakeTupleShape(shape_elements); local
1059 auto result_shape = ShapeUtil::MakeTupleShape( local
1170 auto result_shape = ShapeUtil::MakeShape(S32, {}); local
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/ |
| literal_util.cc | 816 const auto result_shape = local
820 auto result_literal = MakeUnique<Literal>(result_shape);
822 DimensionVector new_indices(ShapeUtil::Rank(result_shape));
823 switch (result_shape.element_type()) {
827 for (int64 i = 0; i < ShapeUtil::Rank(result_shape); ++i) {
837 for (int64 i = 0; i < ShapeUtil::Rank(result_shape); ++i) {
847 for (int64 i = 0; i < ShapeUtil::Rank(result_shape); ++i) {
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tools/parser/ |
| hlo_parser.cc | 312 const Shape& result_shape = computation->root_instruction()->shape(); local
313 if (result_shape.has_layout()) {
316 ->ResetLayout(result_shape.layout());
[all...] |
