Searched refs:op_exec_info (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tfe_src.cc | 1879 FastPathOpExecInfo* op_exec_info) { argument
1880 auto cached_it = op_exec_info->cached_dtypes.find(attr);
1881 if (cached_it != op_exec_info->cached_dtypes.end()) {
1885 auto it = op_exec_info->attr_to_inputs_map->find(attr);
1886 if (it == op_exec_info->attr_to_inputs_map->end()) {
1893 op_exec_info->args, kFastPathExecuteInputStartIndex + input_info.i);
2181 bool CastTensor(const FastPathOpExecInfo& op_exec_info, argument
2190 tensorflow::EagerCast(op_exec_info.ctx, handle->get(), input_dtype,
2202 handle->get(), op_exec_info.ctx, op_exec_info.device_name, status));
2280 const FastPathOpExecInfo& op_exec_info, PyObject* input, argument
[all …]
|