| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_input_output_alias_config.cc | 29 const ShapeIndex& output_index, int64_t param_number, in SetUpAlias() argument
36 TF_RET_CHECK(param_number >= 0) << param_number; in SetUpAlias()
43 param_number, param_index.ToString(), output_index.ToString(), in SetUpAlias()
47 Alias(param_number, param_index, must_alias); in SetUpAlias()
85 int64_t param_number = entry.parameter_number(); in CreateFromProto() local
90 result.SetUpAlias(output_index, param_number, param_index, kind)); in CreateFromProto()
125 int64_t param_number, const ShapeIndex& param_index) const { in ParameterMustAlias() argument
129 if (alias && alias->parameter_number == param_number && in ParameterMustAlias()
138 int64_t param_number, const ShapeIndex& param_index) const { in GetAliasedOutput() argument
142 if (alias && alias->parameter_number == param_number && in GetAliasedOutput()
|
| D | hlo_input_output_alias_config_test.cc | 37 void expect_aliased(const ShapeIndex& output_index, int64_t param_number, in expect_aliased() argument
41 config.GetAliasedOutput(param_number, param_index); in expect_aliased()
50 EXPECT_EQ(aliased_param->parameter_number, param_number); in expect_aliased()
54 void expect_not_aliased(const ShapeIndex& output_index, int64_t param_number, in expect_not_aliased() argument
58 config.GetAliasedOutput(param_number, param_index); in expect_not_aliased()
66 aliased_param->parameter_number == param_number && in expect_not_aliased()
|
| D | hlo_input_output_alias_config.h | 73 Status SetUpAlias(const ShapeIndex& output_index, int64_t param_number,
79 bool ParameterHasAlias(int64_t param_number, in ParameterHasAlias() argument
81 return GetAliasedOutput(param_number, param_index).has_value(); in ParameterHasAlias()
98 int64_t param_number, const ShapeIndex& param_index) const;
108 bool ParameterMustAlias(int64_t param_number,
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | approx_topk.cc | 76 int64_t param_number = 0; in BuildReductionComputation() local
81 lhs_params.push_back(Parameter(reduction_builder.get(), param_number, in BuildReductionComputation()
83 absl::StrFormat("lhs.%d", param_number))); in BuildReductionComputation()
84 param_number++; in BuildReductionComputation()
87 rhs_params.push_back(Parameter(reduction_builder.get(), param_number, in BuildReductionComputation()
89 absl::StrFormat("rhs.%d", param_number))); in BuildReductionComputation()
90 param_number++; in BuildReductionComputation()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | cpu_function_runtime.h | 97 static BufferInfo MakeEntryParameter(uint64_t size, uint64_t param_number) { in MakeEntryParameter() argument
99 /*entry_param_number=*/param_number); in MakeEntryParameter()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_compiler.cc | 548 int64_t param_number, const std::vector<int64_t>& param_index) { in BuildXlaCompilerSubmodule() argument
551 param_number, in BuildXlaCompilerSubmodule()
|
| /external/python/cpython2/Modules/_ctypes/libffi/src/ |
| D | dlmalloc.c | 2595 static int change_mparam(int param_number, int value) { in change_mparam() argument
2598 switch(param_number) { in change_mparam()
4463 int dlmallopt(int param_number, int value) { in dlmallopt() argument
4464 return change_mparam(param_number, value); in dlmallopt()
4889 int mspace_mallopt(int param_number, int value) { in mspace_mallopt() argument
4890 return change_mparam(param_number, value); in mspace_mallopt()
|
| /external/libffi/src/ |
| D | dlmalloc.c | 2595 static int change_mparam(int param_number, int value) { in change_mparam() argument
2598 switch(param_number) { in change_mparam()
4463 int dlmallopt(int param_number, int value) { in dlmallopt() argument
4464 return change_mparam(param_number, value); in dlmallopt()
4889 int mspace_mallopt(int param_number, int value) { in mspace_mallopt() argument
4890 return change_mparam(param_number, value); in mspace_mallopt()
|
| /external/dlmalloc/ |
| D | dlmalloc.c | 3206 static int change_mparam(int param_number, int value) { in change_mparam() argument
3210 switch(param_number) { in change_mparam()
5416 int dlmallopt(int param_number, int value) { in dlmallopt() argument
5417 return change_mparam(param_number, value); in dlmallopt()
6007 int mspace_mallopt(int param_number, int value) { in mspace_mallopt() argument
6008 return change_mparam(param_number, value); in mspace_mallopt()
|
| /external/protobuf/src/google/protobuf/ |
| D | extension_set.h | 107 constexpr ExtensionInfo(const MessageLite* extendee, int param_number, in ExtensionInfo()
111 number(param_number), in ExtensionInfo()
|
| /external/cronet/third_party/protobuf/src/google/protobuf/ |
| D | extension_set.h | 107 constexpr ExtensionInfo(const MessageLite* extendee, int param_number, in ExtensionInfo()
111 number(param_number), in ExtensionInfo()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler.cc | 409 VLOG(3) << "For aliased parameter " << alias.param_number << ": " in BuildComputation()
419 builder->SetUpAlias(alias.output_index, alias.param_number, in BuildComputation()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 398 void SetUpAlias(const ShapeIndex& output_index, int64_t param_number,
403 {output_index, param_number, param_index, kind});
411 int64_t param_number; member
|
| D | xla_builder.cc | 595 if (alias.param_number >= program_shape.parameters_size()) { in PopulateInputOutputAlias()
597 alias.param_number, in PopulateInputOutputAlias()
600 const Shape& parameter_shape = program_shape.parameters(alias.param_number); in PopulateInputOutputAlias()
603 alias.param_number, in PopulateInputOutputAlias()
606 TF_RETURN_IF_ERROR(config.SetUpAlias(alias.output_index, alias.param_number, in PopulateInputOutputAlias()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 3233 int64_t param_number = param_it->second; in EmitThreadLocalBufferPointer() local
3244 params, b_.getInt8PtrTy(), param_number, &b_); in EmitThreadLocalBufferPointer()
|