| /external/tensorflow/tensorflow/python/ops/ |
| D | nccl_ops.py | 33 def all_sum(tensors): argument
79 def all_prod(tensors): argument
96 def all_min(tensors): argument
113 def all_max(tensors): argument
130 def reduce_sum(tensors): argument
210 def _apply_all_reduce(reduction, tensors): argument
239 def _apply_reduce(reduction, tensors): argument
|
| D | nccl_ops_test.py | 33 def _DeviceTensors(tensors, devices): argument
41 def _NcclAllReduce(nccl_fun, tensors, devices): argument
45 def _NcclReduce(nccl_fun, tensors, devices): argument
51 def _NcclBroadcast(tensors, devices): argument
120 def _Gradient(tensors, devices): argument
187 def _Combined(self, tensors, devices): argument
|
| /external/tensorflow/tensorflow/python/data/util/ |
| D | sparse.py | 78 def deserialize_sparse_tensors(tensors, types, shapes, classes): argument
101 def get_classes(tensors): argument
119 def serialize_many_sparse_tensors(tensors): argument
137 def serialize_sparse_tensors(tensors): argument
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | import_test.cc | 104 Offset<Vector<Offset<::tflite::Tensor>>> tensors, in BuildSubGraphs()
125 auto tensors = BuildTensors(); in BuildTestModel() local
148 details::TensorsTable tensors; in TEST_F() local
187 auto tensors = BuildTensors(); in TEST_F() local
201 auto tensors = BuildTensors(); in TEST_F() local
215 auto tensors = BuildTensors(); in TEST_F() local
229 auto tensors = BuildTensors(); in TEST_F() local
244 auto tensors = BuildTensors(); in TEST_F() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_optimizer_test.cc | 80 auto tensors = EvaluateFetchNodes(optimized); in TEST_F() local
139 auto tensors = EvaluateFetchNodes(optimized); in TEST_F() local
188 auto tensors = EvaluateFetchNodes(optimized); in TEST_F() local
245 auto tensors = EvaluateFetchNodes(optimized); in TEST_F() local
292 auto tensors = EvaluateFetchNodes(optimized); in TEST_F() local
340 auto tensors = EvaluateFetchNodes(optimized); in TEST_F() local
542 auto tensors = EvaluateFetchNodes(optimized); in TEST_F() local
571 auto tensors = EvaluateFetchNodes(optimized); in TEST_F() local
729 auto tensors = EvaluateFetchNodes(optimized); in TEST_F() local
989 auto tensors = EvaluateFetchNodes(optimized); in TEST_F() local
[all …]
|
| D | arithmetic_optimizer_test.cc | 132 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
171 auto tensors = EvaluateNodes(output, item.fetch, {{"Placeholder", bool_t}}); in TEST_F() local
210 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
254 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
286 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
332 auto tensors = EvaluateNodes(output, fetch); in TEST_F() local
365 auto tensors = EvaluateNodes(output, fetch); in TEST_F() local
411 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
459 auto tensors = EvaluateNodes(output, fetch); in TEST_F() local
609 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
[all …]
|
| D | pin_to_host_optimizer_test.cc | 86 auto tensors = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
126 auto tensors = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
166 auto tensors = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
202 auto tensors = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
273 auto tensors = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
|
| D | remapper_test.cc | 66 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
105 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
210 auto tensors = EvaluateNodes(output, item.fetch, item.feed); in TEST_F() local
322 auto tensors = EvaluateNodes(output, item.fetch, item.feed); in TEST_F() local
386 auto tensors = EvaluateNodes(output, item.fetch, item.feed); in TEST_F() local
447 auto tensors = EvaluateNodes(output, item.fetch, item.feed); in TEST_F() local
526 auto tensors = EvaluateNodes(output, item.fetch, item.feed); in TEST_F() local
604 auto tensors = EvaluateNodes(output, item.fetch, item.feed); in TEST_F() local
683 auto tensors = EvaluateNodes(output, item.fetch, item.feed); in TEST_F() local
777 auto tensors = EvaluateNodes(output, item.fetch, item.feed); in TEST_F() local
[all …]
|
| D | auto_mixed_precision_test.cc | 170 auto tensors = EvaluateNodes(output, item.fetch, feed); in TestSimpleUnaryGrayOp() local
230 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
268 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
317 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
356 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
401 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
441 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
485 auto tensors = EvaluateNodes(output, item.fetch, feed); in TEST_F() local
542 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
582 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
[all …]
|
| D | constant_folding_test.cc | 120 auto tensors = EvaluateNodes(output, item.fetch, {{"x", x_t}}); in SimpleNeutralElementTest() local
250 auto tensors = EvaluateNodes(output, fetch); in TEST_F() local
341 auto tensors = EvaluateNodes(output, fetch, {{"x", x_t}, {"y", y_t}}); in TEST_F() local
396 auto tensors = EvaluateNodes(output, fetch, {{"x", x_t}}); in TEST_F() local
467 auto tensors = EvaluateNodes(output, fetch, {{"x", x_t}}); in TEST_F() local
543 auto tensors = in TEST_F() local
911 auto tensors = EvaluateNodes( in TEST_F() local
987 auto tensors = EvaluateNodes(output, fetch); in TEST_F() local
1062 auto tensors = EvaluateNodes(output, fetch, in TEST_F() local
1130 auto tensors = EvaluateNodes(output, fetch, in TEST_F() local
[all …]
|
| D | memory_optimizer_test.cc | 292 auto tensors = EvaluateFetchNodes(optimized); in TEST_F() local
342 auto tensors = EvaluateFetchNodes(optimized); in TEST_F() local
391 auto tensors = EvaluateFetchNodes(optimized); in TEST_F() local
437 auto tensors = EvaluateNodes(output, fetch, {}); in TEST_F() local
479 auto tensors = EvaluateFetchNodes(optimized); in TEST_F() local
510 auto tensors = EvaluateFetchNodes(optimized); in TEST_F() local
636 auto tensors = EvaluateFetchNodes(optimized); in TEST_F() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | unique_tensor_references_test.cc | 32 TensorReferenceVector tensors; in TEST() local
55 TensorReferenceVector tensors; in TEST() local
91 TensorReferenceVector tensors; in TEST() local
116 TensorReferenceVector tensors; in TEST() local
|
| D | variant_tensor_data.cc | 29 const Tensor& VariantTensorData::tensors(int index) const { in tensors() function in tensorflow::VariantTensorData
33 const std::vector<Tensor>& VariantTensorData::tensors() const { in tensors() function in tensorflow::VariantTensorData
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | base_layer_utils.py | 161 def have_all_keras_metadata(tensors): argument
169 def create_keras_history(tensors): argument
191 def _create_keras_history_helper(tensors, processed_ops, created_layers): argument
273 def needs_keras_history(tensors, ignore_call_context=False): argument
330 def uses_keras_history(tensors): argument
375 def mark_checked(tensors): argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | tensor_list.h | 126 std::vector<Tensor>& tensors() { return tensors_->values_; } in tensors() function
127 const std::vector<Tensor>& tensors() const { return tensors_->values_; } in tensors() function
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | all_reduce.py | 31 def _flatten_tensors(tensors): argument
60 def _reshape_tensors(tensors, shape): argument
131 def _strip_padding(tensors, pad_len): argument
848 def upper_builder(tensors): argument
851 def upper_level_f(tensors): argument
860 def upper_builder(tensors): argument
863 def upper_level_f(tensors): argument
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | testutil.cc | 27 void GetTensors(const Scope& scope, OutputList tensors, in GetTensors()
40 const OutputList& tensors, std::vector<Tensor>* out) { in GetTensors()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | subscribe.py | 30 def _recursive_apply(tensors, apply_fn): argument
312 def subscribe(tensors, side_effects): argument
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_memory_test.cc | 58 std::set<string> tensors; in TEST_F() local
88 std::set<string> tensors; in TEST_F() local
215 std::set<string> tensors; in TEST_F() local
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | convert_saved_model_test.py | 67 def _getArrayNames(self, tensors): argument
70 def _getArrayShapes(self, tensors): argument
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | forwardprop_util.py | 42 def pack_tangents(tensors): argument
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | pack_test.cc | 41 TfLiteTensor tensors[tensors_size] = { in TestPackTwoInputsFloat() local
116 TfLiteTensor tensors[tensors_size] = { in TestPackThreeInputsFloat() local
189 TfLiteTensor tensors[tensors_size] = { in TestPackTwoInputsQuantized() local
261 TfLiteTensor tensors[tensors_size] = { in TestPackTwoInputsQuantized32() local
|
| D | activations_test.cc | 36 TfLiteTensor tensors[tensors_size] = { in TestReluFloat() local
92 TfLiteTensor tensors[tensors_size] = { in TestRelu6Float() local
151 TfLiteTensor tensors[tensors_size] = { in TestReluUint8() local
216 TfLiteTensor tensors[tensors_size] = { in TestRelu6Uint8() local
280 TfLiteTensor tensors[tensors_size] = { in TestReluInt8() local
346 TfLiteTensor tensors[tensors_size] = { in TestRelu6Int8() local
|
| /external/tensorflow/tensorflow/python/eager/benchmarks/resnet50/ |
| D | resnet50_test.py | 248 def __init__(self, tensors): argument
390 def make_iterator(tensors): argument
401 def make_iterator(tensors): argument
|
| /external/tensorflow/tensorflow/python/training/ |
| D | input.py | 445 def _as_tensor_list(tensors): argument
760 def _batch(tensors, batch_size, keep_input, num_threads=1, capacity=32, argument
835 def _shuffle_batch(tensors, batch_size, capacity, min_after_dequeue, argument
929 def batch(tensors, batch_size, num_threads=1, capacity=32, argument
1028 def maybe_batch(tensors, keep_input, batch_size, num_threads=1, capacity=32, argument
1251 def shuffle_batch(tensors, batch_size, capacity, min_after_dequeue, argument
1355 def maybe_shuffle_batch(tensors, batch_size, capacity, min_after_dequeue, argument
|