Searched refs:all_tensors (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/testing/op_tests/ |
| D | pack.py | 76 all_tensors = []
82 all_tensors.append(input_tensor)
83 out = tf.stack(all_tensors, parameters["axis"])
84 return all_tensors, [out]
|
| D | concat.py | 83 all_tensors = []
89 all_tensors.append(input_tensor)
90 out = tf.concat(all_tensors, parameters["axis"])
91 return all_tensors, [out]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | custom_gradient.py | 392 all_tensors = flat_result + args + variables
417 original_tensors = all_tensors
419 all_tensors = array_ops.identity_n(all_tensors)
426 all_tensors[i]._handle_data = t._handle_data # pylint: disable=protected-access
428 f.__name__, all_tensors, original_tensors, tape_grad_fn)
429 for ot, t in zip(original_tensors, all_tensors):
432 structure=result, flat_sequence=all_tensors[:flat_result_len])
|
| D | math_ops.py | 223 all_tensors = (expanded_start, res, expanded_stop)
224 concatenated = array_ops.concat(all_tensors, axis=axis)
|
| /external/tensorflow/tensorflow/python/tools/ |
| D | inspect_checkpoint.py | 57 def print_tensors_in_checkpoint_file(file_name, tensor_name, all_tensors, argument
76 if all_tensors or all_tensor_names:
81 if all_tensors:
159 FLAGS.all_tensors, FLAGS.all_tensor_names,
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | gradient_input_output_exclusions.py | 327 all_tensors = set(range(num_values))
328 unused_tensors = all_tensors - used_tensors
|
| /external/tensorflow/tensorflow/python/training/ |
| D | saver.py | 144 all_tensors = []
151 all_tensors.extend(
153 return all_tensors
335 all_tensors = self.bulk_restore(filename_tensor, saveables, preferred_shard,
356 saveable_tensors = all_tensors[idx:idx + len(saveable.specs)]
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 1846 std::set<nvinfer1::ITensor*> all_tensors; in MaybeApplyQuantizationRanges() local
1851 all_tensors.insert(layer->getInput(j)); in MaybeApplyQuantizationRanges()
1855 all_tensors.insert(layer->getOutput(j)); in MaybeApplyQuantizationRanges()
1870 all_tensors.insert(input_tensor); in MaybeApplyQuantizationRanges()
1944 for (auto tensor : all_tensors) { in MaybeApplyQuantizationRanges()
|