Searched refs:MakeEntryParameter (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/aot/ |
| D | codegen_test.cc | 219 BufferInfo::MakeEntryParameter(/*size=*/8, /*param_number=*/0), in TEST()
221 BufferInfo::MakeEntryParameter(/*size=*/96, /*param_number=*/1), in TEST()
223 BufferInfo::MakeEntryParameter(/*size=*/96, /*param_number=*/2), in TEST()
225 BufferInfo::MakeEntryParameter(/*size=*/96, /*param_number=*/3), in TEST()
227 BufferInfo::MakeEntryParameter(/*size=*/96, /*param_number=*/4), in TEST()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | cpu_function_runtime_test.cc | 166 BufferInfo::MakeEntryParameter(/*size=*/0, /*param_number=*/4)); in TEST()
168 BufferInfo::MakeEntryParameter(/*size=*/4, /*param_number=*/0)); in TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | buffer_info_util.cc | 33 buffer_infos.push_back(BufferInfo::MakeEntryParameter( in CreateBufferInfosFromBufferAssignment()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | cpu_function_runtime.h | 96 static BufferInfo MakeEntryParameter(tensorflow::uint64 size, in MakeEntryParameter() function
|