Home
last modified time | relevance | path

Searched refs:param_number (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_input_output_alias_config.cc28 int64 param_number, in SetUpAlias() argument
37 TF_RET_CHECK(param_number >= 0) << param_number; in SetUpAlias()
44 param_number, param_index.ToString(), output_index.ToString(), in SetUpAlias()
48 Alias(kind, param_number, param_index); in SetUpAlias()
91 int64 param_number = entry.parameter_number(); in CreateFromProto() local
101 result.SetUpAlias(output_index, param_number, param_index, kind)); in CreateFromProto()
126 int64 param_number, const ShapeIndex& param_index) const { in ParameterAliasKind() argument
130 if (alias && alias->parameter_number == param_number && in ParameterAliasKind()
139 int64 param_number, const ShapeIndex& param_index) const { in GetAliasedOutput() argument
143 if (alias && alias->parameter_number == param_number && in GetAliasedOutput()
Dhlo_input_output_alias_config.h66 Status SetUpAlias(const ShapeIndex& output_index, int64 param_number,
71 AliasKind ParameterAliasKind(int64 param_number,
76 bool ParameterHasAlias(int64 param_number, in ParameterHasAlias() argument
78 return ParameterAliasKind(param_number, param_index) != AliasKind::kNoAlias; in ParameterHasAlias()
95 int64 param_number, const ShapeIndex& param_index) const;
Dhlo_input_output_alias_config_test.cc37 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()
Doptimize_input_output_buffer_alias.cc53 int param_number; in Build() member
84 output_index, entry.param_number, input_index, in Build()
/external/tensorflow/tensorflow/compiler/xla/
Dcpu_function_runtime.h97 tensorflow::uint64 param_number) { in MakeEntryParameter() argument
99 /*entry_param_number=*/param_number); in MakeEntryParameter()
/external/u-boot/common/
Ddlmalloc.c2361 int mALLOPt(int param_number, int value) in mALLOPt() argument
2363 int mALLOPt(param_number, value) int param_number; int value; in mALLOPt()
2366 switch(param_number)
Ddlmalloc.src3135 int mALLOPt(int param_number, int value)
3137 int mALLOPt(param_number, value) int param_number; int value;
3140 switch(param_number)
/external/python/cpython2/Modules/_ctypes/libffi/src/
Ddlmalloc.c2595 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/
Ddlmalloc.c2590 static int change_mparam(int param_number, int value) { in change_mparam() argument
2593 switch(param_number) { in change_mparam()
4458 int dlmallopt(int param_number, int value) { in dlmallopt() argument
4459 return change_mparam(param_number, value); in dlmallopt()
4884 int mspace_mallopt(int param_number, int value) { in mspace_mallopt() argument
4885 return change_mparam(param_number, value); in mspace_mallopt()
/external/dlmalloc/
Dmalloc.c3206 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/xla/client/
Dxla_builder.h335 void SetUpAlias(const ShapeIndex& output_index, int64 param_number, in SetUpAlias() argument
337 input_output_aliases_.push_back({output_index, param_number, param_index}); in SetUpAlias()
345 int64 param_number; member
Dxla_builder.cc453 if (alias.param_number >= program_shape.parameters_size()) { in PopulateInputOutputAlias()
455 alias.param_number, in PopulateInputOutputAlias()
458 const Shape& parameter_shape = program_shape.parameters(alias.param_number); in PopulateInputOutputAlias()
461 alias.param_number, in PopulateInputOutputAlias()
465 alias.output_index, alias.param_number, alias.param_index, in PopulateInputOutputAlias()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler.cc1349 builder.SetUpAlias(alias.output_index, alias.param_number, in CompileGraph()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc3161 int64 param_number = param_it->second; in EmitThreadLocalBufferPointer() local
3172 llvm_ir::EmitBufferIndexingGEP(params, param_number, &b_); in EmitThreadLocalBufferPointer()