| /external/tensorflow/tensorflow/compiler/xla/service/ |
| tuple_points_to_analysis_test.cc | 518 // Bitcast is an alias of its operand. A tuple with a bitcast element should
519 // have the operand of the bitcast in its points-to set.
525 auto bitcast = builder.AddInstruction(HloInstruction::CreateUnary( local
528 builder.AddInstruction(HloInstruction::CreateTuple({constant1, bitcast}));
532 EXPECT_EQ(1, points_to_analysis_->GetPointsToSet(bitcast).size());
534 points_to_analysis_->GetPointsToSet(bitcast).element({}), {constant2});
536 points_to_analysis_->GetPointsToSet(bitcast).tuple_sources({}).empty());
[all...] |
| hlo_alias_analysis_test.cc | 810 auto bitcast = builder.AddInstruction(HloInstruction::CreateUnary( local
829 auto bitcast = builder.AddInstruction(HloInstruction::CreateUnary( local
[all...] |
| copy_insertion_test.cc | 228 // The output of a bitcast is its operand (same buffer), so a bitcast
233 HloInstruction* bitcast = builder.AddInstruction(HloInstruction::CreateUnary( local
239 EXPECT_THAT(x->users(), UnorderedElementsAre(bitcast));
250 // The output of a bitcast is its operand (same buffer), so a bitcast
255 HloInstruction* bitcast = builder.AddInstruction(HloInstruction::CreateUnary( local
261 EXPECT_THAT(constant->users(), UnorderedElementsAre(bitcast));
272 // Same as BitcastParameter, but the bitcast is wrapped in a tuple.
276 HloInstruction* bitcast = builder.AddInstruction(HloInstruction::CreateUnary local
[all...] |
| hlo_dataflow_analysis_test.cc | 1093 auto bitcast = builder.AddInstruction(HloInstruction::CreateUnary( local
[all...] |
| algebraic_simplifier.cc | 66 // identical to its operand and thus may be replaced with a bitcast.
75 // identical to its operand and thus may be replaced with a bitcast.
78 // reshape may still be a bitcast. For example, a reshape from [28x28] to [784].
123 Status HandleBitcast(HloInstruction* bitcast) override;
223 // Convenience method for replacing an instruction with a bitcast.
297 // Callback used to determine if a bitcast is possible.
335 auto bitcast = computation_->AddInstruction( local
338 TF_CHECK_OK(ReplaceInstruction(instruction, bitcast));
400 Status AlgebraicSimplifierVisitor::HandleBitcast(HloInstruction* bitcast) {
401 // If a bitcast feeds a bitcast, make it a single bitcast
[all...] |
| buffer_assignment_test.cc | 1327 auto bitcast = builder.AddInstruction( local
[all...] |
| /external/mesa3d/src/gallium/drivers/radeonsi/ |
| si_shader_tgsi_setup.c | 284 LLVMValueRef bitcast(struct lp_build_tgsi_context *bld_base, function
528 bitcast(bld_base, TGSI_TYPE_UNSIGNED, ptr),
532 bitcast(bld_base, TGSI_TYPE_UNSIGNED, ptr2),
534 return bitcast(bld_base, type, result);
689 return bitcast(bld_base, type, load);
762 return bitcast(bld_base, type, result);
778 return bitcast(bld_base, type, cval);
1021 value = bitcast(bld_base, TGSI_TYPE_FLOAT, value);
1061 LLVMBuildStore(builder, bitcast(bld_base, TGSI_TYPE_FLOAT, value), temp_ptr);
1062 LLVMBuildStore(builder, bitcast(bld_base, TGSI_TYPE_FLOAT, val2), temp_ptr2)
[all...] |
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| arithmetic_optimizer.cc | 216 if (node.op() == "Bitcast") {
226 if (node.op() == "Bitcast") {
706 if (node->op() == "Bitcast") {
707 NodeDef* bitcast = node_map_->GetNode(node->name()); local
709 if (GetSourceDataType(*bitcast) == GetDestinationDataType(*bitcast)) {
710 return bitcast->input(0);
713 const NodeDef* operand = node_map_->GetNode(bitcast->input(0));
714 if (operand->op() == bitcast->op()) {
715 // Bitcast(Bitcast(x, type1), type2) => Bitcast(x, type2
[all...] |
