Home
last modified time | relevance | path

Searched refs:is_entry_parameter (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dcpu_function_runtime.h47 bool is_entry_parameter() const { return kind() == Kind::kEntryParameter; } in is_entry_parameter() function
51 assert(is_entry_parameter()); in entry_parameter_number()
81 return !is_entry_parameter() ||
Dcpu_function_runtime.cc64 (buffer_infos[i].is_entry_parameter() && allocate_entry_params); in AlignedBufferBytes()
91 (buffer_infos[i].is_entry_parameter() && allocate_entry_params); in MallocContiguousBuffers()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dbuffer_info_util.cc46 if (buffer_infos[i].is_entry_parameter()) { in CreateArgIndexTableFromBufferInfos()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dlocal_client_aot_test_helper.cc96 CHECK(result->buffer_infos()[0].is_entry_parameter()); // param buffer in main()
/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_assignment.cc1006 const bool is_entry_parameter = in AssignBuffersForComputation() local
1009 if (is_entry_parameter) { in AssignBuffersForComputation()
1114 CHECK(!is_entry_parameter) << *buffer; in AssignBuffersForComputation()
/external/tensorflow/tensorflow/compiler/aot/
Dcodegen.cc107 return buffer_info.is_entry_parameter(); in ExtractEntryParamBufferInfos()