Searched refs:flat_result (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/python/eager/ |
D | custom_gradient.py | 55 flat_result = nest.flatten(result) 56 all_tensors = flat_result + args 60 gradients = nest.flatten(grad_fn(*result_grads[:len(flat_result)])) 64 return ([None] * len(flat_result)) + gradients 70 structure=result, flat_sequence=all_tensors[:len(flat_result)]) 76 flat_result = nest.flatten(result) 78 flat_result = [gen_array_ops.identity(x) for x in flat_result] 85 flat_result, 88 flat_result = list(flat_result) 89 return nest.pack_sequence_as(result, flat_result)
|
D | backprop.py | 695 flat_result = nest.flatten(result) 696 flat_result = [gen_array_ops.identity(x) for x in flat_result] 697 result = nest.pack_sequence_as(result, flat_result)
|
D | pywrap_tfe_src.cc | 1739 PyObject* flat_result = PyList_New(num_retvals); local 1741 PyList_SET_ITEM(flat_result, i, EagerTensorFromHandle(retvals[i])); 1746 args, *flattened_inputs, *flattened_attrs, flat_result, 1759 return flat_result; 1761 auto* result = PyList_GET_ITEM(flat_result, 0); 1763 Py_DECREF(flat_result); 1776 PyObject* obj = PyList_GET_ITEM(flat_result, flat_result_index++); 1785 PyObject* obj = PyList_GET_ITEM(flat_result, flat_result_index++); 1791 PyObject* obj = PyList_GET_ITEM(flat_result, flat_result_index++); 1796 Py_DECREF(flat_result);
|
/external/tensorflow/tensorflow/core/kernels/ |
D | quantization_utils.h | 733 auto flat_result = result->flat<T>(); in FloatTensorToQuantizedInPlaceUsingEigen() local 734 DCHECK_EQ(flat_input.size(), flat_result.size()); in FloatTensorToQuantizedInPlaceUsingEigen() 737 flat_result.device(device) = QUANTIZE_WITH_EIGEN(flat_input, f2q, T); in FloatTensorToQuantizedInPlaceUsingEigen() 745 auto flat_result = result->flat<T>(); in FloatTensorToQuantizedInPlace() local 747 DCHECK(data_size == flat_result.size()); in FloatTensorToQuantizedInPlace() 749 flat_result(i) = FloatToQuantized<T>(flat_input(i), min, max); in FloatTensorToQuantizedInPlace() 767 auto flat_result = result->flat<float>(); in QuantizedTensorToFloatInPlaceUsingEigen() local 769 DCHECK(data_size == flat_result.size()); in QuantizedTensorToFloatInPlaceUsingEigen() 772 flat_result.device(device) = DEQUANTIZE_WITH_EIGEN(flat_input, q2f); in QuantizedTensorToFloatInPlaceUsingEigen() 781 auto flat_result = result->flat<float>(); in QuantizedTensorToFloatInPlace() local [all …]
|
/external/tensorflow/tensorflow/python/ops/ |
D | rnn.py | 318 for r, flat_result in zip(result, flat_results): 320 flat_result.append(r) 322 results = [nest.pack_sequence_as(structure=input_, flat_sequence=flat_result) 323 for input_, flat_result in zip(input_seq, flat_results)]
|
D | control_flow_ops.py | 3011 flat_result = nest.flatten(original_body_result) 3015 _convert_flows_to_tensorarrays(flat_result, exit_vars))
|