/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_input_output_alias_config.cc | 44 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 …]
|
D | sort_simplifier.cc | 85 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()
|
D | hlo_input_output_alias_config.h | 45 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()
|
D | source_map_util.h | 48 Status InvalidParameterArgument(Executable* executable, int parameter_number, in InvalidParameterArgument() argument 54 HloInstruction* param = computation.parameter_instruction(parameter_number); in InvalidParameterArgument()
|
D | topk_rewriter.cc | 29 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()
|
D | local_service.cc | 81 const XlaComputation& computation, int parameter_number) { in ParameterMetadata() argument 86 instr.parameter_number() == parameter_number) { in ParameterMetadata()
|
D | map_inliner.cc | 83 map->ReplaceAllUsesWith(map->operands()[root.parameter_number()])); in HandleMap() 100 params.push_back(map->operands()[root.operand(o)->parameter_number()]); in HandleMap()
|
D | copy_insertion.cc | 49 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()
|
D | hlo_input_output_alias_config_test.cc | 50 EXPECT_EQ(aliased_param->parameter_number, param_number); in expect_aliased() 66 aliased_param->parameter_number == param_number && in expect_not_aliased()
|
D | hlo_matchers.h | 46 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()
|
D | stable_sort_expander_test.cc | 43 return a->parameter_number() == (b->parameter_number() ^ 1); in IsSameComputationExceptParams()
|
D | hlo_matchers.cc | 87 if (instruction->parameter_number() != parameter_number_) { in MatchAndExplain() 89 << instruction->parameter_number() << ", want " in MatchAndExplain()
|
D | while_loop_all_reduce_code_motion.cc | 126 int parameter_number = in IsAllReduceMovable() local 127 Cast<HloParameterInstruction>(instruction)->parameter_number(); in IsAllReduceMovable() 128 CHECK_EQ(parameter_number, 0); in IsAllReduceMovable()
|
D | buffer_assignment.h | 127 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()
|
D | memory_space_propagation.cc | 93 parent_fusion->operand(instruction->parameter_number()); in Propagate()
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | alias_passthrough_params.cc | 34 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()
|
D | alias_passthrough_params_test.cc | 40 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/ |
D | client_library_test_base.h | 280 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 …]
|
D | client_library_test_base.cc | 609 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/ |
D | utils.cc | 85 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/ |
D | tpu_executable_interface.cc | 71 (*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/ |
D | cpu_executable.cc | 97 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()
|
D | buffer_info_util.cc | 35 /*param_number=*/allocation.parameter_number())); in CreateBufferInfosFromBufferAssignment()
|
D | cpu_layout_assignment_test.cc | 322 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/ |
D | simple.hlo | 37 parameter_number: 1
|