Home
last modified time | relevance | path

Searched refs:first_operand (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cse_test.cc102 auto first_operand = add->operand(0); in TEST_F() local
103 EXPECT_THAT(first_operand, ::testing::AnyOf(constant1, constant2)); in TEST_F()
104 EXPECT_THAT(add, op::Add(first_operand, first_operand)); in TEST_F()
215 auto first_operand = tuple->operand(0); in TEST_F() local
216 EXPECT_THAT(first_operand, in TEST_F()
219 op::Tuple(first_operand, first_operand, uncommon_constant)); in TEST_F()
246 auto first_operand = tuple->operand(0); in TEST_F() local
247 EXPECT_THAT(first_operand, ::testing::AnyOf(exp1, exp2, exp3)); in TEST_F()
248 EXPECT_THAT(tuple, op::Tuple(first_operand, first_operand, first_operand)); in TEST_F()
483 auto first_operand = tuple->operand(0); in TEST_F() local
[all …]
Dspace_to_batch_converter.cc828 auto first_operand = old_to_new_instrs_[activations]; in CanPropagate() local
830 instr_to_dim_permute_map_[first_operand]; in CanPropagate()
835 const int64 lhs_batch = first_operand->shape().dimensions(new_batch_dim); in CanPropagate()
837 if (first_operand->shape().dimensions(new_space_dim) % rhs_dilation != in CanPropagate()
884 auto first_operand = old_to_new_instrs_[activations]; in CanPropagate() local
889 auto permute_dims_first_operand = instr_to_dim_permute_map_[first_operand]; in CanPropagate()
903 if (first_operand->shape().dimensions(new_batch_dim_operand_0) != in CanPropagate()
910 if (first_operand->shape().dimensions(new_space_dim_operand_0) > in CanPropagate()
948 auto first_operand = old_to_new_instrs_[consumer->mutable_operand(0)]; in CanPropagate() local
952 auto permute_dims_first_operand = instr_to_dim_permute_map_[first_operand]; in CanPropagate()
[all …]
/external/deqp-deps/SPIRV-Tools/source/opt/
Deliminate_dead_members_pass.cpp177 uint32_t first_operand = (inst->opcode() == SpvOpSpecConstantOp ? 1 : 0); in MarkMembersAsLiveForExtract() local
178 uint32_t composite_id = inst->GetSingleWordInOperand(first_operand); in MarkMembersAsLiveForExtract()
182 for (uint32_t i = first_operand + 1; i < inst->NumInOperands(); ++i) { in MarkMembersAsLiveForExtract()
538 uint32_t first_operand = 0; in UpdateCompsiteExtract() local
540 first_operand = 1; in UpdateCompsiteExtract()
542 uint32_t object_id = inst->GetSingleWordInOperand(first_operand); in UpdateCompsiteExtract()
548 for (uint32_t i = 0; i < first_operand + 1; i++) { in UpdateCompsiteExtract()
551 for (uint32_t i = first_operand + 1; i < inst->NumInOperands(); ++i) { in UpdateCompsiteExtract()
593 uint32_t first_operand = 0; in UpdateCompositeInsert() local
595 first_operand = 1; in UpdateCompositeInsert()
[all …]
Dvector_dce.cpp172 WorkListItem first_operand; in MarkVectorShuffleUsesAsLive() local
173 first_operand.instruction = in MarkVectorShuffleUsesAsLive()
181 type_mgr->GetType(first_operand.instruction->type_id())->AsVector(); in MarkVectorShuffleUsesAsLive()
189 first_operand.components.Set(index); in MarkVectorShuffleUsesAsLive()
196 AddItemToWorkListIfNeeded(first_operand, live_components, work_list); in MarkVectorShuffleUsesAsLive()
/external/swiftshader/third_party/SPIRV-Tools/source/opt/
Deliminate_dead_members_pass.cpp177 uint32_t first_operand = (inst->opcode() == SpvOpSpecConstantOp ? 1 : 0); in MarkMembersAsLiveForExtract() local
178 uint32_t composite_id = inst->GetSingleWordInOperand(first_operand); in MarkMembersAsLiveForExtract()
182 for (uint32_t i = first_operand + 1; i < inst->NumInOperands(); ++i) { in MarkMembersAsLiveForExtract()
538 uint32_t first_operand = 0; in UpdateCompsiteExtract() local
540 first_operand = 1; in UpdateCompsiteExtract()
542 uint32_t object_id = inst->GetSingleWordInOperand(first_operand); in UpdateCompsiteExtract()
548 for (uint32_t i = 0; i < first_operand + 1; i++) { in UpdateCompsiteExtract()
551 for (uint32_t i = first_operand + 1; i < inst->NumInOperands(); ++i) { in UpdateCompsiteExtract()
593 uint32_t first_operand = 0; in UpdateCompositeInsert() local
595 first_operand = 1; in UpdateCompositeInsert()
[all …]
Dvector_dce.cpp172 WorkListItem first_operand; in MarkVectorShuffleUsesAsLive() local
173 first_operand.instruction = in MarkVectorShuffleUsesAsLive()
181 type_mgr->GetType(first_operand.instruction->type_id())->AsVector(); in MarkVectorShuffleUsesAsLive()
189 first_operand.components.Set(index); in MarkVectorShuffleUsesAsLive()
196 AddItemToWorkListIfNeeded(first_operand, live_components, work_list); in MarkVectorShuffleUsesAsLive()