/external/tensorflow/tensorflow/compiler/xla/service/ |
H A D | computation_layout.h | 65 const Shape& result_shape() const { return result_layout_.shape(); } function in class:xla::ComputationLayout
|
H A D | executable.h | 151 const Shape& result_shape() const { function in class:xla::Executable 152 return hlo_module_->config().entry_computation_layout().result_shape();
|
H A D | 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);
|
H A D | 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, { [all...] |
H A D | 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 local [all...] |
H A D | shape_inference_test.cc | 1224 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 D | algebraic_simplifier_test.cc | 1091 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 D | 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));
|
H A D | reshape_util.cc | 139 Tensor *result_shape = nullptr; local 142 &result_shape)); 143 auto output_shape_vec = result_shape->vec<int64>();
|
H A D | 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));
|
H A D | 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/ |
H A D | bcast.h | 99 const Vec& result_shape() const { return result_; } function in class:tensorflow::BCast
|
H A D | tensor_slice_set.cc | 35 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 D | 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_lh 219 Shape result_shape = ShapeUtil::MakeShape(F32, {1, 24}); local [all...] |
H A D | 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 local 418 Shape result_shape = ShapeUtil::MakeShape(F32, {4, 3}); local [all...] |
/external/tensorflow/tensorflow/compiler/xla/tests/ |
H A D | reduce_window_test.cc | 330 Shape result_shape = local 332 std::unique_ptr<Literal> expected = Literal::CreateFromShape(result_shape);
|
H A D | 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 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 D | 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) { 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 D | hlo_parser.cc | 312 const Shape& result_shape = computation->root_instruction()->shape(); local 313 if (result_shape.has_layout()) { 316 ->ResetLayout(result_shape.layout());
|