Searched refs:skip_input_indices (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | math_grad.py | 1283 skip_input_indices = None
1285 skip_input_indices = op.skip_input_indices
1286 if skip_input_indices is not None and 1 in skip_input_indices and _IsScalar(
1298 if skip_input_indices is not None and 0 in skip_input_indices:
1304 if skip_input_indices is not None and 1 in skip_input_indices:
1317 skip_input_indices = None
1319 skip_input_indices = op.skip_input_indices
1320 if skip_input_indices is not None and 1 in skip_input_indices and _IsScalar(
1332 if skip_input_indices is not None and 0 in skip_input_indices:
1338 if skip_input_indices is not None and 1 in skip_input_indices:
[all …]
|
| D | while_v2.py | 341 if "skip_input_indices" in op.__dict__ and op.skip_input_indices is not None:
344 for i in op.skip_input_indices:
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | backprop.py | 106 def __init__(self, attrs, inputs, outputs, typ, skip_input_indices): argument
111 self.skip_input_indices = skip_input_indices
130 out_grads, skip_input_indices, forward_pass_name_scope): argument
147 mock_op = _MockOp(attr_tuple, inputs, outputs, op_name, skip_input_indices)
|
| D | pywrap_tfe_src.cc | 3248 tensorflow::Safe_PyObjectPtr skip_input_indices; local
3250 skip_input_indices.reset(
3254 skip_input_indices.get(), i,
3259 skip_input_indices.reset(Py_None);
3263 output_grads, skip_input_indices.get(),
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | gradients.h | 70 std::vector<int64> skip_input_indices; member
|