Home
last modified time | relevance | path

Searched refs:parameter_number (Results 1 – 25 of 69) 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_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 …]
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()
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 parameter_number; member
58 return absl::StrFormat("(%lld, %s, %s)", parameter_number, in ToString()
Dtopk_rewriter.cc29 auto match_bitcast_f32 = [](int64_t parameter_number) { in IsNanSafeGt() argument
30 auto param = m::Parameter(parameter_number) in IsNanSafeGt()
45 auto match_bitcast_f32_with_convert = [](int64_t parameter_number) { in IsNanSafeGt() argument
46 auto param = m::Parameter(parameter_number) in IsNanSafeGt()
61 auto match_bitcast_bf16 = [](int64_t parameter_number) { in IsNanSafeGt() argument
62 auto param = m::Convert(m::Parameter(parameter_number) in IsNanSafeGt()
78 auto match_bitcast_bf16_with_convert = [](int64_t parameter_number) { in IsNanSafeGt() argument
79 auto param = m::Convert(m::Parameter(parameter_number) in IsNanSafeGt()
Dsource_map_util.h48 Status InvalidParameterArgument(Executable* executable, int parameter_number, in InvalidParameterArgument() argument
54 HloInstruction* param = computation.parameter_instruction(parameter_number); in InvalidParameterArgument()
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()
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.h49 explicit HloParameterMatcher(int64_t parameter_number) in HloParameterMatcher() argument
51 parameter_number_(parameter_number) {}
325 int64_t parameter_number) { in Parameter() argument
327 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()
Dbuffer_assignment.h127 int64 parameter_number() const { in parameter_number() function
278 void set_entry_computation_parameter(int64_t parameter_number, in set_entry_computation_parameter() argument
283 parameter_number_ = parameter_number; in set_entry_computation_parameter()
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()
Dcopy_insertion.cc54 value.defining_instruction()->parameter_number(), value.index()); in IsReadonlyEntryParameterValue()
372 if (alias.parameter_number == param->parameter_number()) { in AddCopiesForAliasedInputOutputs()
384 TF_RET_CHECK(param->parameter_number() < entry->num_parameters()); in AddCopiesForAliasedInputOutputs()
385 TF_RET_CHECK(!copied_parameters[param->parameter_number()]); in AddCopiesForAliasedInputOutputs()
400 copied_parameters[param->parameter_number()] = param_copy_tree; in AddCopiesForAliasedInputOutputs()
419 if (!copied_parameters[alias.parameter_number]) { in AddCopiesForAliasedInputOutputs()
423 copied_parameters[alias.parameter_number]->element( in AddCopiesForAliasedInputOutputs()
1884 << alias->parameter_number << " copying " << other_index; in AddSpecialCaseCopies()
Dmemory_space_propagation.cc98 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()
43 root->operand(i)->parameter_number(), /*param_index=*/{})) { in Run()
51 /*param_number=*/root->operand(i)->parameter_number(), in Run()
53 used_params.insert(root->operand(i)->parameter_number()); in Run()
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()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dclient_library_test_base.h280 int64_t parameter_number, const Literal& literal, const string& name,
287 int64_t parameter_number, const Literal& literal, const string& name,
336 int64_t parameter_number,
351 absl::Span<const NativeT> values, int64_t parameter_number,
365 const Array2D<NativeT>& array_2d, int64_t parameter_number,
379 const Array3D<NativeT>& array_3d, int64_t parameter_number,
393 const Array4D<NativeT>& array_4d, int64_t parameter_number,
398 int64_t parameter_number,
606 NativeT value, int64_t 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.cc610 int64_t parameter_number, const Literal& literal, const string& name, in CreateParameterAndTransferLiteral() argument
612 return CreateParameterAndTransferLiteral(parameter_number, literal, name, in CreateParameterAndTransferLiteral()
640 int64_t parameter_number, const Literal& literal, const string& name, in CreateParameterAndTransferLiteral() argument
647 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()
232 if (alias.parameter_number != 0) { in ComputeParametersThatMustBeDonated()
236 alias.parameter_number); in ComputeParametersThatMustBeDonated()
250 int this_parameter = alias.parameter_number; in ComputeParametersThatMustBeDonated()
/external/tensorflow/tensorflow/stream_executor/tpu/
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()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_executable.cc98 se::DeviceMemoryBase out = arguments[allocation.parameter_number()] in MemoryForAllocation()
102 << "Size mismatch on param " << allocation.parameter_number() in MemoryForAllocation()
265 CHECK_LT(alias->parameter_number, arguments.size()); in CreateResultShapedBuffer()
266 ExecutionInput& input = arguments[alias->parameter_number]; in CreateResultShapedBuffer()
Dbuffer_info_util.cc35 /*param_number=*/allocation.parameter_number())); in CreateBufferInfosFromBufferAssignment()
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()
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Dsimple.hlo37 parameter_number: 1

123