Searched defs:result_shape (Results 1 - 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
H A Dcomputation_layout.h65 const Shape& result_shape() const { return result_layout_.shape(); } function in class:xla::ComputationLayout
H A Dexecutable.h151 const Shape& result_shape() const { function in class:xla::Executable
152 return hlo_module_->config().entry_computation_layout().result_shape();
H A Dhlo_module.cc297 const Shape& result_shape = local
299 TF_RET_CHECK(ShapeUtil::Equal(expected_program_shape.result(), result_shape))
303 << ", actual: " << ShapeUtil::HumanStringWithLayout(result_shape);
H A Dlayout_assignment_test.cc245 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, {
[all...]
H A Dlayout_assignment.cc558 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 local
[all...]
H A Dshape_inference_test.cc1224 Shape result_shape = ShapeUtil::MakeTupleShape({s32_, vector_32_}); local
1225 ProgramShape cond = ShapeUtil::MakeProgramShape({result_shape}, pred_);
1226 ProgramShape body = ShapeUtil::MakeProgramShape({result_shape}, result_shape);
1228 ShapeInference::InferWhileShape(cond, body, result_shape);
1231 ASSERT_TRUE(ShapeUtil::Equal(result_shape, inferred));
1236 Shape result_shape = ShapeUtil::MakeTupleShape({s32_, vector_32_}); local
1237 ProgramShape cond = ShapeUtil::MakeProgramShape({result_shape}, pred_);
1238 ProgramShape body = ShapeUtil::MakeProgramShape({result_shape}, result_shape);
[all...]
H A Dalgebraic_simplifier_test.cc1091 Shape result_shape = ShapeUtil::MakeShape(F32, {3 * kParamLength}); local
1093 result_shape, {empty_literal, param0, param0, empty_slice, param1}, 0));
1121 Shape result_shape = ShapeUtil::MakeShape(F32, {0}); local
1123 result_shape, {empty_literal, empty_slice}, 0));
/external/tensorflow/tensorflow/core/kernels/
H A Dgather_op.cc89 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));
H A Dreshape_util.cc139 Tensor *result_shape = nullptr; local
142 &result_shape));
143 auto output_shape_vec = result_shape->vec<int64>();
H A Ddynamic_stitch_op.cc127 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));
H A Dresource_variable_ops.cc506 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/
H A Dbcast.h99 const Vec& result_shape() const { return result_; } function in class:tensorflow::BCast
H A Dtensor_slice_set.cc35 TensorShape result_shape; local
36 TF_RETURN_IF_ERROR(slice.SliceTensorShape(shape_, &result_shape));
57 result_shape.num_elements()};
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_layout_assignment_test.cc61 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_lh
219 Shape result_shape = ShapeUtil::MakeShape(F32, {1, 24}); local
[all...]
H A Dcpu_instruction_fusion_test.cc250 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 local
418 Shape result_shape = ShapeUtil::MakeShape(F32, {4, 3}); local
[all...]
/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dreduce_window_test.cc330 Shape result_shape = local
332 std::unique_ptr<Literal> expected = Literal::CreateFromShape(result_shape);
H A Dwhile_test.cc54 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 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/
H A Dliteral_util.cc816 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) {
857 for (int64 i = 0; i < ShapeUtil::Rank(result_shape); ++i) {
866 << PrimitiveType_Name(result_shape.element_type());
/external/tensorflow/tensorflow/compiler/xla/tools/parser/
H A Dhlo_parser.cc312 const Shape& result_shape = computation->root_instruction()->shape(); local
313 if (result_shape.has_layout()) {
316 ->ResetLayout(result_shape.layout());

Completed in 1004 milliseconds