Searched refs:all_tensors (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | custom_gradient.py | 213 all_tensors = flat_result + args + variables
238 original_tensors = all_tensors
240 all_tensors = array_ops.identity_n(all_tensors)
247 all_tensors[i]._handle_data = t._handle_data # pylint: disable=protected-access
249 f.__name__, all_tensors, original_tensors, tape_grad_fn)
250 for ot, t in zip(original_tensors, all_tensors):
253 structure=result, flat_sequence=all_tensors[:len(flat_result)])
|
| /external/tensorflow/tensorflow/python/tools/ |
| D | inspect_checkpoint.py | 32 def print_tensors_in_checkpoint_file(file_name, tensor_name, all_tensors, argument
49 if all_tensors or all_tensor_names:
53 if all_tensors:
118 FLAGS.all_tensors, FLAGS.all_tensor_names)
|
| /external/tensorflow/tensorflow/python/training/ |
| D | saver.py | 146 all_tensors = []
153 all_tensors.extend(
155 return all_tensors
330 all_tensors = self.bulk_restore(filename_tensor, saveables, preferred_shard,
351 saveable_tensors = all_tensors[idx:idx + len(saveable.specs)]
|
| /external/tensorflow/tensorflow/contrib/graph_editor/ |
| D | subgraph.py | 203 all_tensors = frozenset(inputs + outputs + list(insides))
204 self._passthrough_ts = [t for t in passthrough_ts if t not in all_tensors]
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | generate_examples.py | 1971 all_tensors = []
1976 all_tensors.append(input_tensor)
1977 out = tf.concat(all_tensors, parameters["axis"])
1978 return all_tensors, [out]
3915 all_tensors = []
3921 all_tensors.append(input_tensor)
3922 out = tf.stack(all_tensors, parameters["axis"])
3923 return all_tensors, [out]
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 1322 std::set<nvinfer1::ITensor*> all_tensors; in MaybeApplyQuantizationRanges() local
1326 all_tensors.insert(layer->getInput(j)); in MaybeApplyQuantizationRanges()
1329 all_tensors.insert(layer->getOutput(j)); in MaybeApplyQuantizationRanges()
1333 for (auto tensor : all_tensors) { in MaybeApplyQuantizationRanges()
|