| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_input_output_alias_config.cc | 29 const ShapeIndex& output_index, int64 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 param_number = entry.parameter_number(); in CreateFromProto() local
90 result.SetUpAlias(output_index, param_number, param_index, kind)); in CreateFromProto()
125 int64 param_number, const ShapeIndex& param_index) const { in ParameterMustAlias() argument
129 if (alias && alias->parameter_number == param_number && in ParameterMustAlias()
138 int64 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.h | 73 Status SetUpAlias(const ShapeIndex& output_index, int64 param_number,
79 bool ParameterHasAlias(int64 param_number, in ParameterHasAlias() argument
81 return GetAliasedOutput(param_number, param_index).has_value(); in ParameterHasAlias()
98 int64 param_number, const ShapeIndex& param_index) const;
108 bool ParameterMustAlias(int64 param_number,
|
| D | hlo_input_output_alias_config_test.cc | 37 void expect_aliased(const ShapeIndex& output_index, int64 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 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()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | cpu_function_runtime.h | 97 tensorflow::uint64 param_number) { in MakeEntryParameter() argument
99 /*entry_param_number=*/param_number); in MakeEntryParameter()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_compiler.cc | 409 int64 param_number, const std::vector<int64>& param_index) { in BuildXlaCompilerSubmodule() argument
412 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 | malloc.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/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler.cc | 405 VLOG(3) << "For aliased parameter " << alias.param_number << ": " in BuildComputation()
415 builder->SetUpAlias(alias.output_index, alias.param_number, in BuildComputation()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 401 void SetUpAlias(const ShapeIndex& output_index, int64 param_number,
406 {output_index, param_number, param_index, kind});
414 int64 param_number; member
|
| D | xla_builder.cc | 532 if (alias.param_number >= program_shape.parameters_size()) { in PopulateInputOutputAlias()
534 alias.param_number, in PopulateInputOutputAlias()
537 const Shape& parameter_shape = program_shape.parameters(alias.param_number); in PopulateInputOutputAlias()
540 alias.param_number, in PopulateInputOutputAlias()
543 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 | 3032 int64 param_number = param_it->second; in EmitThreadLocalBufferPointer() local
3043 llvm_ir::EmitBufferIndexingGEP(params, param_number, &b_); in EmitThreadLocalBufferPointer()
|