Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/compiler/xla/service/
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 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 …]
Dsort_simplifier.cc85 int64 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()
Dhlo_input_output_alias_config.h45 Alias(int64 parameter_number, ShapeIndex parameter_index,
47 : parameter_number(parameter_number), in parameter_number() function
51 int64 parameter_number; member
58 return absl::StrFormat("(%lld, %s, %s)", parameter_number, in ToString()
Dsource_map_util.h48 Status InvalidParameterArgument(Executable* executable, int parameter_number, in InvalidParameterArgument() argument
54 HloInstruction* param = computation.parameter_instruction(parameter_number); in InvalidParameterArgument()
Dtopk_rewriter.cc29 auto match_bitcast_f32 = [](int64 parameter_number) { in IsNanSafeGt() argument
30 auto param = m::Parameter(parameter_number) in IsNanSafeGt()
44 auto match_bitcast_bf16 = [](int64 parameter_number) { in IsNanSafeGt() argument
45 auto param = m::Convert(m::Parameter(parameter_number) in IsNanSafeGt()
Dlocal_service.cc81 const XlaComputation& computation, int parameter_number) { in ParameterMetadata() argument
86 instr.parameter_number() == parameter_number) { in ParameterMetadata()
Dmap_inliner.cc83 map->ReplaceAllUsesWith(map->operands()[root.parameter_number()])); in HandleMap()
100 params.push_back(map->operands()[root.operand(o)->parameter_number()]); in HandleMap()
Dcopy_insertion.cc49 value.defining_instruction()->parameter_number(), value.index()); in IsReadonlyEntryParameterValue()
367 if (alias.parameter_number == param->parameter_number()) { in AddCopiesForAliasedInputOutputs()
379 TF_RET_CHECK(param->parameter_number() < entry->num_parameters()); in AddCopiesForAliasedInputOutputs()
380 TF_RET_CHECK(!copied_parameters[param->parameter_number()]); in AddCopiesForAliasedInputOutputs()
395 copied_parameters[param->parameter_number()] = param_copy_tree; in AddCopiesForAliasedInputOutputs()
414 if (!copied_parameters[alias.parameter_number]) { in AddCopiesForAliasedInputOutputs()
418 copied_parameters[alias.parameter_number]->element( in AddCopiesForAliasedInputOutputs()
1115 << alias->parameter_number << " copying " << other_index; in AddSpecialCaseCopies()
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()
Dhlo_matchers.h46 explicit HloParameterMatcher(int64 parameter_number) in HloParameterMatcher() argument
48 parameter_number_(parameter_number) {}
291 int64 parameter_number) { in Parameter() argument
293 new ::xla::testing::HloParameterMatcher(parameter_number)); in Parameter()
Dstable_sort_expander_test.cc43 return a->parameter_number() == (b->parameter_number() ^ 1); in IsSameComputationExceptParams()
Dhlo_matchers.cc87 if (instruction->parameter_number() != parameter_number_) { in MatchAndExplain()
89 << instruction->parameter_number() << ", want " in MatchAndExplain()
Dwhile_loop_all_reduce_code_motion.cc126 int parameter_number = in IsAllReduceMovable() local
127 Cast<HloParameterInstruction>(instruction)->parameter_number(); in IsAllReduceMovable()
128 CHECK_EQ(parameter_number, 0); in IsAllReduceMovable()
Dbuffer_assignment.h127 int64 parameter_number() const { in parameter_number() function
277 void set_entry_computation_parameter(int64 parameter_number, in set_entry_computation_parameter() argument
282 parameter_number_ = parameter_number; in set_entry_computation_parameter()
Dmemory_space_propagation.cc93 parent_fusion->operand(instruction->parameter_number()); in Propagate()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dalias_passthrough_params.cc34 used_params.count(root->operand(i)->parameter_number()) == 0) { in Run()
35 VLOG(2) << "Parameter " << root->operand(i)->parameter_number() in Run()
42 /*param_number=*/root->operand(i)->parameter_number(), in Run()
44 used_params.insert(root->operand(i)->parameter_number()); in Run()
Dalias_passthrough_params_test.cc40 EXPECT_EQ(0, alias_config.GetAliasedParameter({0})->parameter_number); in TEST_F()
42 EXPECT_EQ(1, alias_config.GetAliasedParameter({2})->parameter_number); in TEST_F()
56 EXPECT_EQ(0, alias_config.GetAliasedParameter({0})->parameter_number); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dclient_library_test_base.h280 int64 parameter_number, const Literal& literal, const string& name,
287 int64 parameter_number, const Literal& literal, const string& name,
336 int64 parameter_number,
351 absl::Span<const NativeT> values, int64 parameter_number,
365 const Array2D<NativeT>& array_2d, int64 parameter_number,
379 const Array3D<NativeT>& array_3d, int64 parameter_number,
393 const Array4D<NativeT>& array_4d, int64 parameter_number,
398 int64 parameter_number,
606 NativeT value, int64 parameter_number, const string& name, in CreateR0Parameter() argument
614 *data_handle = Parameter(builder, parameter_number, literal.shape(), name); in CreateR0Parameter()
[all …]
Dclient_library_test_base.cc609 ClientLibraryTestBase::CreateParameterAndTransferLiteral(int64 parameter_number, in CreateParameterAndTransferLiteral() argument
614 return CreateParameterAndTransferLiteral(parameter_number, literal, name, in CreateParameterAndTransferLiteral()
642 int64 parameter_number, const Literal& literal, const string& name, in CreateParameterAndTransferLiteral() argument
649 Parameter(builder, parameter_number, param_literal.shape(), name); in CreateParameterAndTransferLiteral()
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dutils.cc85 if (instr.parameter_number() >= program_shape.parameters_size()) { in GetShardedProgramShapes()
88 instr.parameter_number(), program_shape.parameters_size()); in GetShardedProgramShapes()
90 TF_ASSIGN_OR_RETURN(arg_shapes[instr.parameter_number()], in GetShardedProgramShapes()
231 if (alias.parameter_number != 0) { in GetParametersThatMustBeDonated()
235 alias.parameter_number); in GetParametersThatMustBeDonated()
249 int this_parameter = alias.parameter_number; in GetParametersThatMustBeDonated()
/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_executable_interface.cc71 (*arguments)[alias->parameter_number].Buffers().element( in AllocateOutputMemoryWithInputReuse()
115 TF_RET_CHECK(alias->parameter_number < arguments->size()); in AllocateOutputMemoryWithInputReuse()
116 ExecutionInput& input = (*arguments)[alias->parameter_number]; in AllocateOutputMemoryWithInputReuse()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_executable.cc97 se::DeviceMemoryBase out = arguments[allocation.parameter_number()] in MemoryForAllocation()
101 << "Size mismatch on param " << allocation.parameter_number() in MemoryForAllocation()
262 CHECK_LT(alias->parameter_number, arguments.size()); in CreateResultShapedBuffer()
263 ExecutionInput& input = arguments[alias->parameter_number]; in CreateResultShapedBuffer()
Dbuffer_info_util.cc35 /*param_number=*/allocation.parameter_number())); in CreateBufferInfosFromBufferAssignment()
Dcpu_layout_assignment_test.cc322 fusion_instruction->operand(fused_dot->operand(0)->parameter_number()); in RunDotOutputFusion()
324 fusion_instruction->operand(fused_dot->operand(1)->parameter_number()); in RunDotOutputFusion()
326 fused_add->operand(1 - dot_operand_idx_in_add)->parameter_number()); in RunDotOutputFusion()
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Dsimple.hlo37 parameter_number: 1

123