Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_input_output_alias_config.cc27 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(kind, param_number, param_index); in SetUpAlias()
90 int64 param_number = entry.parameter_number(); in CreateFromProto() local
100 result.SetUpAlias(output_index, param_number, param_index, kind)); in CreateFromProto()
121 int64 param_number, const ShapeIndex& param_index) const { in ParameterAliasKind() argument
125 if (alias && alias->parameter_number == param_number && in ParameterAliasKind()
134 int64 param_number, const ShapeIndex& param_index) const { in GetAliasedOutput() argument
138 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.cc38 void expect_aliased(const ShapeIndex& output_index, int64 param_number, in expect_aliased() argument
42 config.GetAliasedOutput(param_number, param_index); in expect_aliased()
51 EXPECT_EQ(aliased_param->parameter_number, param_number); in expect_aliased()
55 void expect_not_aliased(const ShapeIndex& output_index, int64 param_number, in expect_not_aliased() argument
59 config.GetAliasedOutput(param_number, param_index); in expect_not_aliased()
67 aliased_param->parameter_number == param_number && in expect_not_aliased()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dcpu_function_runtime.h95 static BufferInfo MakeEntryParameter(uint64 size, uint64 param_number) { in MakeEntryParameter() argument
97 /*entry_param_number=*/param_number); in MakeEntryParameter()
Dxla_compiler.cc1098 builder.SetUpAlias(alias.output_index, alias.param_number, in CompileGraph()
/external/u-boot/common/
Ddlmalloc.c2362 int mALLOPt(int param_number, int value) in mALLOPt() argument
2364 int mALLOPt(param_number, value) int param_number; int value; in mALLOPt()
2367 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.h302 void SetUpAlias(const ShapeIndex& output_index, int64 param_number, in SetUpAlias() argument
304 input_output_aliases_.push_back({output_index, param_number, param_index}); in SetUpAlias()
312 int64 param_number; member
Dxla_builder.cc399 if (alias.param_number >= program_shape.parameters_size()) { in PopulateInputOutputAlias()
401 alias.param_number, in PopulateInputOutputAlias()
404 const Shape& parameter_shape = program_shape.parameters(alias.param_number); in PopulateInputOutputAlias()
407 alias.param_number, in PopulateInputOutputAlias()
411 alias.output_index, alias.param_number, alias.param_index, in PopulateInputOutputAlias()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc2851 int64 param_number = param_it->second; in EmitThreadLocalBufferPointer() local
2862 llvm_ir::EmitBufferIndexingGEP(params, param_number, &b_); in EmitThreadLocalBufferPointer()