Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/ops/
Dcustom_gradient.py212 flat_result = nest.flatten(result)
213 all_tensors = flat_result + args + variables
217 result_grads = result_grads[:len(flat_result)]
231 return ([None] * len(flat_result)) + input_grads + variable_grads
253 structure=result, flat_sequence=all_tensors[:len(flat_result)])
270 flat_result = nest.flatten(result)
272 flat_result = [gen_array_ops.identity(x) for x in flat_result]
294 tape_lib.record_operation(f.__name__, flat_result, input_tensors,
296 flat_result = list(flat_result)
297 return nest.pack_sequence_as(result, flat_result)
Drnn.py342 for r, flat_result in zip(result, flat_results):
344 flat_result.append(r)
346 results = [nest.pack_sequence_as(structure=input_, flat_sequence=flat_result)
347 for input_, flat_result in zip(input_seq, flat_results)]
Darray_ops.py3517 flat_result = gather(flat_params, flat_indices)
3518 result = reshape(flat_result, concat([indices_shape, outer_shape], axis=0))
Dcontrol_flow_ops.py3002 flat_result = nest.flatten(original_body_result, expand_composites=True)
3006 _convert_flows_to_tensorarrays(flat_result, exit_vars))
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_string_ops.py376 flat_result = decode_op(
384 codepoints = flat_result.char_values
386 offsets = flat_result.char_to_byte_starts
389 flat_result.char_values, flat_result.row_splits)
394 flat_result.char_to_byte_starts, flat_result.row_splits)
/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/eager/
Dbackprop.py526 flat_result = nest.flatten(result)
527 flat_result = [gen_array_ops.identity(x) for x in flat_result]
528 result = nest.pack_sequence_as(result, flat_result)
Dpywrap_tfe_src.cc2785 tensorflow::Safe_PyObjectPtr flat_result(PyList_New(num_retvals)); local
2787 PyList_SET_ITEM(flat_result.get(), i, EagerTensorFromHandle(retvals[i]));
2791 flattened_attrs.get(), flat_result.get())) {
2803 return flat_result.release();
2805 auto* result = PyList_GET_ITEM(flat_result.get(), 0);
2819 PyObject* obj = PyList_GET_ITEM(flat_result.get(), flat_result_index++);
2828 PyObject* obj = PyList_GET_ITEM(flat_result.get(), flat_result_index++);
2834 PyObject* obj = PyList_GET_ITEM(flat_result.get(), flat_result_index++);