Home
last modified time | relevance | path

Searched refs:parameter_number (Results 1 – 25 of 75) sorted by relevance

123

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_input_output_alias_config.cc44 alias_.element(output_index)->parameter_number, in SetUpAlias()
63 entry.set_parameter_number(data->parameter_number); in ToProto()
85 int64_t param_number = entry.parameter_number(); in CreateFromProto()
107 alias.parameter_number, alias.parameter_index.ToString())); in ToString()
129 if (alias && alias->parameter_number == param_number && in ParameterMustAlias()
142 if (alias && alias->parameter_number == param_number && in GetAliasedOutput()
191 TF_RET_CHECK(0 <= alias.parameter_number); in Verify()
192 TF_RET_CHECK(entry->num_parameters() > alias.parameter_number); in Verify()
194 entry->parameter_instruction(alias.parameter_number)->shape(); in Verify()
211 alias.parameter_number, alias.parameter_index.ToString(), in Verify()
[all …]
H A Dtopk_rewriter.cc30 auto match_bitcast_f32 = [](int64_t parameter_number) { in IsNanSafeGt() argument
31 auto param = m::Parameter(parameter_number) in IsNanSafeGt()
46 auto match_bitcast_f32_with_convert = [](int64_t parameter_number) { in IsNanSafeGt() argument
47 auto param = m::Parameter(parameter_number) in IsNanSafeGt()
62 auto match_bitcast_bf16 = [](int64_t parameter_number) { in IsNanSafeGt() argument
63 auto param = m::Convert(m::Parameter(parameter_number) in IsNanSafeGt()
79 auto match_bitcast_bf16_with_convert = [](int64_t parameter_number) { in IsNanSafeGt() argument
80 auto param = m::Convert(m::Parameter(parameter_number) in IsNanSafeGt()
96 auto match_s32 = [](int64_t parameter_number) { in IsNanSafeGt() argument
97 auto param = m::Parameter(parameter_number) in IsNanSafeGt()
H A Dsort_simplifier.cc85 int64_t parameter_number = 0; in RemoveUnusedOperandFromSort() local
93 parameter_number, scalar_shape, in RemoveUnusedOperandFromSort()
94 absl::StrCat("p.", parameter_number / 2, ".lhs")); in RemoveUnusedOperandFromSort()
95 ++parameter_number; in RemoveUnusedOperandFromSort()
97 parameter_number, scalar_shape, in RemoveUnusedOperandFromSort()
98 absl::StrCat("p.", parameter_number / 2, ".rhs")); in RemoveUnusedOperandFromSort()
99 ++parameter_number; in RemoveUnusedOperandFromSort()
H A Dhlo_input_output_alias_config.h45 Alias(int64_t parameter_number, ShapeIndex parameter_index,
47 : parameter_number(parameter_number), in parameter_number() function
51 int64_t parameter_number; member
58 return absl::StrFormat("(%lld, %s, %s)", parameter_number, in ToString()
H A Dlocal_service.cc80 const XlaComputation& computation, int parameter_number) { in ParameterMetadata() argument
85 instr.parameter_number() == parameter_number) { in ParameterMetadata()
H A Dsource_map_util.h47 Status InvalidParameterArgument(Executable* executable, int parameter_number, in InvalidParameterArgument() argument
53 HloInstruction* param = computation.parameter_instruction(parameter_number); in InvalidParameterArgument()
H A Dhlo_dataflow_analysis.cc909 callsite.instruction()->operand(parameter->parameter_number()))); in UpdateParameterValueSet()
914 CHECK_EQ(parameter->parameter_number(), 0); in UpdateParameterValueSet()
927 CHECK_EQ(parameter->parameter_number(), 0); in UpdateParameterValueSet()
949 callsite.instruction()->operand(parameter->parameter_number()))); in UpdateParameterValueSet()
954 {0, parameter->parameter_number()}); in UpdateParameterValueSet()
1268 for (int64_t parameter_number = 0; in Propagate() local
1269 parameter_number < in Propagate()
1271 ++parameter_number) { in Propagate()
1274 parameter_number)); in Propagate()
1758 HloOperandIndex{in_place_input_source->parameter_number(), in GetFusionInstructionInPlaceInputOutputPairs()
H A Dmap_inliner.cc82 map->ReplaceAllUsesWith(map->operands()[root.parameter_number()])); in HandleMap()
99 params.push_back(map->operands()[root.operand(o)->parameter_number()]); in HandleMap()
H A Dcopy_insertion.cc58 value.defining_instruction()->parameter_number(), value.index()); in IsReadonlyEntryParameterValue()
377 if (alias.parameter_number == param->parameter_number()) { in AddCopiesForAliasedInputOutputs()
389 TF_RET_CHECK(param->parameter_number() < entry->num_parameters()); in AddCopiesForAliasedInputOutputs()
390 TF_RET_CHECK(!copied_parameters[param->parameter_number()]); in AddCopiesForAliasedInputOutputs()
409 copied_parameters[param->parameter_number()] = param_copy_tree; in AddCopiesForAliasedInputOutputs()
428 if (!copied_parameters[alias.parameter_number]) { in AddCopiesForAliasedInputOutputs()
432 copied_parameters[alias.parameter_number]->element( in AddCopiesForAliasedInputOutputs()
1960 << alias->parameter_number << " copying " << other_index; in AddSpecialCaseCopies()
H A Dhlo_input_output_alias_config_test.cc50 EXPECT_EQ(aliased_param->parameter_number, param_number); in expect_aliased()
66 aliased_param->parameter_number == param_number && in expect_not_aliased()
H A Dhlo_matchers.h49 explicit HloParameterMatcher(int64_t parameter_number) in HloParameterMatcher() argument
51 parameter_number_(parameter_number) {}
342 int64_t parameter_number) { in Parameter() argument
344 new ::xla::testing::HloParameterMatcher(parameter_number)); in Parameter()
H A Dbuffer_assignment.h125 int64_t parameter_number() const { in parameter_number() function
276 void set_entry_computation_parameter(int64_t parameter_number, in set_entry_computation_parameter() argument
281 parameter_number_ = parameter_number; in set_entry_computation_parameter()
H A Dstable_sort_expander_test.cc43 return a->parameter_number() == (b->parameter_number() ^ 1); in IsSameComputationExceptParams()
H A Dhlo_matchers.cc89 if (instruction->parameter_number() != parameter_number_) { in MatchAndExplain()
91 << instruction->parameter_number() << ", want " in MatchAndExplain()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dclient_library_test_base.h278 int64_t parameter_number, const Literal& literal, const std::string& name,
285 int64_t parameter_number, const Literal& literal, const std::string& name,
334 int64_t parameter_number,
349 absl::Span<const NativeT> values, int64_t parameter_number,
363 const Array2D<NativeT>& array_2d, int64_t parameter_number,
377 const Array3D<NativeT>& array_3d, int64_t parameter_number,
391 const Array4D<NativeT>& array_4d, int64_t parameter_number,
396 int64_t parameter_number,
606 NativeT value, int64_t parameter_number, const std::string& name, in CreateR0Parameter() argument
613 *data_handle = Parameter(builder, parameter_number, literal.shape(), name); in CreateR0Parameter()
[all …]
H A Dclient_library_test_base.cc612 int64_t parameter_number, const Literal& literal, const std::string& name, in CreateParameterAndTransferLiteral() argument
614 return CreateParameterAndTransferLiteral(parameter_number, literal, name, in CreateParameterAndTransferLiteral()
642 int64_t parameter_number, const Literal& literal, const std::string& name, in CreateParameterAndTransferLiteral() argument
649 Parameter(builder, parameter_number, param_literal.shape(), name); in CreateParameterAndTransferLiteral()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dalias_passthrough_params.cc36 used_params.count(root->operand(i)->parameter_number()) == 0) { in Run()
37 VLOG(2) << "Parameter " << root->operand(i)->parameter_number() in Run()
45 root->operand(i)->parameter_number(), /*param_index=*/{})) { in Run()
53 /*param_number=*/root->operand(i)->parameter_number(), in Run()
55 used_params.insert(root->operand(i)->parameter_number()); in Run()
H A Dalias_passthrough_params_test.cc41 EXPECT_EQ(0, alias_config.GetAliasedParameter({0})->parameter_number); in TEST_F()
43 EXPECT_EQ(1, alias_config.GetAliasedParameter({2})->parameter_number); in TEST_F()
57 EXPECT_EQ(0, alias_config.GetAliasedParameter({0})->parameter_number); in TEST_F()
83 EXPECT_EQ(1, alias_result.GetAliasedParameter({2})->parameter_number); in TEST_F()
H A Dfusion_bitcast_lift.cc154 int64_t parameter_number = i->parameter_number(); in Run() local
156 auto n = HloInstruction::CreateParameter(parameter_number, in Run()
159 i->parent()->ReplaceParameter(parameter_number, std::move(n)); in Run()
171 cloned_fusion->mutable_operand(parameter_number); in Run()
177 parameter_number, new_op)); in Run()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dutils.cc87 if (instr.parameter_number() >= program_shape.parameters_size()) { in GetShardedProgramShapes()
90 instr.parameter_number(), program_shape.parameters_size()); in GetShardedProgramShapes()
92 TF_ASSIGN_OR_RETURN(arg_shapes[instr.parameter_number()], in GetShardedProgramShapes()
234 if (alias.parameter_number != 0) { in ComputeParametersThatMustBeDonated()
238 alias.parameter_number); in ComputeParametersThatMustBeDonated()
252 int this_parameter = alias.parameter_number; in ComputeParametersThatMustBeDonated()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_executable.cc100 se::DeviceMemoryBase out = arguments[allocation.parameter_number()] in MemoryForAllocation()
104 << "Size mismatch on param " << allocation.parameter_number() in MemoryForAllocation()
255 CHECK_LT(alias->parameter_number, arguments.size()); in CreateResultShapedBuffer()
256 ExecutionInput& input = arguments[alias->parameter_number]; in CreateResultShapedBuffer()
H A Dcpu_layout_assignment_test.cc321 fusion_instruction->operand(fused_dot->operand(0)->parameter_number()); in RunDotOutputFusion()
323 fusion_instruction->operand(fused_dot->operand(1)->parameter_number()); in RunDotOutputFusion()
325 fused_add->operand(1 - dot_operand_idx_in_add)->parameter_number()); in RunDotOutputFusion()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
H A Dtpu_executable_interface.cc83 (*arguments)[alias->parameter_number].Buffers().element( in AllocateOutputMemoryWithInputReuse()
127 TF_RET_CHECK(alias->parameter_number < arguments->size()); in AllocateOutputMemoryWithInputReuse()
128 ExecutionInput& input = (*arguments)[alias->parameter_number]; in AllocateOutputMemoryWithInputReuse()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
H A Dexecutable_base.cc164 (*arguments)[alias->parameter_number].Buffers().element( in AllocateOutputMemoryWithInputReuse()
197 TF_RET_CHECK(alias->parameter_number < arguments->size()); in AllocateOutputMemoryWithInputReuse()
198 ExecutionInput& input = (*arguments)[alias->parameter_number]; in AllocateOutputMemoryWithInputReuse()
/aosp_15_r20/external/sdv/vsomeip/third_party/boost/container/src/
Ddlmalloc.cpp61 BOOST_CONTAINER_DECL int dlmalloc_mallopt(int parameter_number, int parameter_value) in dlmalloc_mallopt() argument
62 { return boost_cont_mallopt(parameter_number, parameter_value); } in dlmalloc_mallopt()

123