| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_input_output_alias_config.cc | 44 alias_.element(output_index)->parameter_number, in SetUpAlias()
73 entry.set_parameter_number(data->parameter_number); in ToProto()
90 int64 param_number = entry.parameter_number(); in CreateFromProto()
113 output_index.ToString(), kind, alias.parameter_number, in ToString()
125 if (alias && alias->parameter_number == param_number && in ParameterAliasKind()
138 if (alias && alias->parameter_number == param_number && in GetAliasedOutput()
186 TF_RET_CHECK(0 <= alias.parameter_number); in Verify()
187 TF_RET_CHECK(entry->num_parameters() > alias.parameter_number); in Verify()
189 entry->parameter_instruction(alias.parameter_number)->shape(); in Verify()
206 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 | 47 Alias(AliasKind kind, int64 parameter_number, ShapeIndex parameter_index) in Alias()
49 parameter_number(parameter_number), in Alias()
53 int64 parameter_number; member
|
| 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 | local_service.cc | 80 const XlaComputation& computation, int parameter_number) { in ParameterMetadata() argument
85 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()
355 if (alias.parameter_number == param->parameter_number()) { in AddCopiesForAliasedInputOutputs()
367 TF_RET_CHECK(param->parameter_number() < entry->num_parameters()); in AddCopiesForAliasedInputOutputs()
368 TF_RET_CHECK(!copied_parameters[param->parameter_number()]); in AddCopiesForAliasedInputOutputs()
383 copied_parameters[param->parameter_number()] = param_copy_tree; in AddCopiesForAliasedInputOutputs()
402 if (!copied_parameters[alias.parameter_number]) { in AddCopiesForAliasedInputOutputs()
406 copied_parameters[alias.parameter_number]->element( in AddCopiesForAliasedInputOutputs()
|
| D | hlo_input_output_alias_config_test.cc | 51 EXPECT_EQ(aliased_param->parameter_number, param_number); in expect_aliased()
67 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) {}
260 int64 parameter_number) { in Parameter() argument
262 new ::xla::testing::HloParameterMatcher(parameter_number)); in Parameter()
|
| D | hlo_matchers.cc | 83 if (instruction->parameter_number() != parameter_number_) { in MatchAndExplain()
85 << instruction->parameter_number() << ", want " in MatchAndExplain()
|
| D | stable_sort_expander_test.cc | 43 return a->parameter_number() == (b->parameter_number() ^ 1); in IsSameComputationExceptParams()
|
| D | buffer_assignment.h | 124 int64 parameter_number() const { in parameter_number() function
280 void set_entry_computation_parameter(int64 parameter_number, in set_entry_computation_parameter() argument
285 parameter_number_ = parameter_number; in set_entry_computation_parameter()
|
| D | hlo.proto | 63 int64 parameter_number = 9; field
267 // parameter_number=0,
277 int64 parameter_number = 2; field
384 int64 parameter_number = 6; field
|
| D | call_inliner.cc | 72 parameter, call_->mutable_operand(parameter->parameter_number()))); in HandleParameter()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | fused_ir_emitter.h | 104 llvm_ir::IrArray& GetIrArrayForFusedParameter(int64 parameter_number) { in GetIrArrayForFusedParameter() argument
108 return operand_arrays_.value()[parameter_number]; in GetIrArrayForFusedParameter()
111 llvm::Value* GetBasePointerForFusedParameter(int64 parameter_number) { in GetBasePointerForFusedParameter() argument
112 return GetIrArrayForFusedParameter(parameter_number).GetBasePointer(); in GetBasePointerForFusedParameter()
|
| D | fused_ir_emitter.cc | 120 GetBasePointerForFusedParameter(tuple_operand->parameter_number()); in HandleGetTupleElement()
150 parameter->parameter_number())) { in HandleParameter()
163 return GetIrArrayForFusedParameter(parameter->parameter_number()) in HandleParameter()
|
| D | dynamic_update_slice_util.h | 60 auto* operand = fusion->operand(fusion_operand->parameter_number()); in CanEmitFusedDynamicUpdateSliceInPlace()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | client_library_test_base.h | 272 int64 parameter_number, const Literal& literal, const string& name,
279 int64 parameter_number, const Literal& literal, const string& name,
328 int64 parameter_number,
343 absl::Span<const NativeT> values, int64 parameter_number,
357 const Array2D<NativeT>& array_2d, int64 parameter_number,
371 const Array3D<NativeT>& array_3d, int64 parameter_number,
550 NativeT value, int64 parameter_number, const string& name, in CreateR0Parameter() argument
558 *data_handle = Parameter(builder, parameter_number, literal.shape(), name); in CreateR0Parameter()
564 absl::Span<const NativeT> values, int64 parameter_number, in CreateR1Parameter() argument
572 *data_handle = Parameter(builder, parameter_number, literal.shape(), name); in CreateR1Parameter()
[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
650 Parameter(builder, parameter_number, param_literal.shape(), name); in CreateParameterAndTransferLiteral()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_executable.cc | 93 unowning_buffers[i] = arguments[allocation.parameter_number()]->buffer( in CreateBufferTable()
253 slice.allocation()->parameter_number(), in CreateResultShapedBuffer()
257 << slice.allocation()->parameter_number() << " at index " in CreateResultShapedBuffer()
|
| D | buffer_info_util.cc | 34 /*param_number=*/allocation.parameter_number())); in CreateBufferInfosFromBufferAssignment()
|
| D | cpu_layout_assignment_test.cc | 315 fusion_instruction->operand(fused_dot->operand(0)->parameter_number()); in RunDotOutputFusion()
317 fusion_instruction->operand(fused_dot->operand(1)->parameter_number()); in RunDotOutputFusion()
319 fused_add->operand(1 - dot_operand_idx_in_add)->parameter_number()); in RunDotOutputFusion()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_executable.cc | 255 auto param_no = allocation.parameter_number(); in ExecuteOnStream()
332 slice.allocation()->parameter_number(), in ExecuteOnStream()
336 << slice.allocation()->parameter_number() << " at index " in ExecuteOnStream()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_execute_op.cc | 241 TF_RET_CHECK(alias.parameter_number < input_tuples.size()); in DoWork()
244 *input_tuples[alias.parameter_number], in DoWork()
|
| /external/clang/test/SemaCXX/ |
| D | addr-of-overloaded-function.cpp | 182 void parameter_number() { in parameter_number() function
|