HomeSort by relevance Sort by last modified time
    Searched full:flat_result (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/python/eager/
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
    [all...]
pywrap_tfe_src.cc 1739 PyObject* flat_result = PyList_New(num_retvals); local
    [all...]
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)
  /external/tensorflow/tensorflow/core/kernels/
quantization_utils.h     [all...]
  /external/tensorflow/tensorflow/python/ops/
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)]
    [all...]
control_flow_ops.py     [all...]

Completed in 2493 milliseconds