Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/eager/
Dcustom_gradient.py55 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)
Dbackprop.py695 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)
Dpywrap_tfe_src.cc1739 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/
Dquantization_utils.h733 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/
Drnn.py318 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)]
Dcontrol_flow_ops.py3011 flat_result = nest.flatten(original_body_result)
3015 _convert_flows_to_tensorarrays(flat_result, exit_vars))