Home
last modified time | relevance | path

Searched defs:tensors (Results 1 – 25 of 271) sorted by relevance

1234567891011

/external/tensorflow/tensorflow/python/ops/
Dnccl_ops.py33 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
Dnccl_ops_test.py33 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/security/advisory/
Dtfsa-2021-103.md3 ### CVE Number
6 ### Impact
12 ### Patches
20 ### For more information
/external/tensorflow/tensorflow/python/data/util/
Dsparse.py78 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/armnn/src/armnnTfLiteParser/test/
DGetInputsOutputs.cpp156 TfLiteParser::TensorRawPtrVector tensors = TfLiteParser::GetInputs(model, 0, 0); in BOOST_FIXTURE_TEST_CASE() local
163 TfLiteParser::TensorRawPtrVector tensors = TfLiteParser::GetOutputs(model, 0, 0); in BOOST_FIXTURE_TEST_CASE() local
170 TfLiteParser::TensorRawPtrVector tensors = TfLiteParser::GetInputs(model, 0, 0); in BOOST_FIXTURE_TEST_CASE() local
179 TfLiteParser::TensorRawPtrVector tensors = TfLiteParser::GetOutputs(model, 0, 0); in BOOST_FIXTURE_TEST_CASE() local
188 TfLiteParser::TensorRawPtrVector tensors = TfLiteParser::GetInputs(model, 1, 0); in BOOST_FIXTURE_TEST_CASE() local
199 TfLiteParser::TensorRawPtrVector tensors = TfLiteParser::GetOutputs(model, 1, 0); in BOOST_FIXTURE_TEST_CASE() local
/external/tensorflow/tensorflow/lite/toco/tflite/
Dimport_test.cc105 Offset<Vector<Offset<::tflite::Tensor>>> tensors, in BuildSubGraphs()
126 auto tensors = BuildTensors(); in BuildTestModel() local
149 details::TensorsTable tensors; in TEST_F() local
188 auto tensors = BuildTensors(); in TEST_F() local
202 auto tensors = BuildTensors(); in TEST_F() local
216 auto tensors = BuildTensors(); in TEST_F() local
230 auto tensors = BuildTensors(); in TEST_F() local
245 auto tensors = BuildTensors(); in TEST_F() local
/external/armnn/src/armnnOnnxParser/test/
DGetInputsOutputs.cpp72 std::vector<std::string> tensors = armnnOnnxParser::OnnxParser::GetInputs(model); in BOOST_FIXTURE_TEST_CASE() local
81 std::vector<std::string> tensors = armnnOnnxParser::OnnxParser::GetOutputs(model); in BOOST_FIXTURE_TEST_CASE() local
143 std::vector<std::string> tensors = armnnOnnxParser::OnnxParser::GetInputs(model); in BOOST_FIXTURE_TEST_CASE() local
247 std::vector<std::string> tensors = armnnOnnxParser::OnnxParser::GetInputs(model); in BOOST_FIXTURE_TEST_CASE() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_optimizer_test.cc80 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 …]
Darithmetic_optimizer_test.cc385 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
606 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
652 auto tensors = EvaluateNodes(output, fetch); in TEST_F() local
685 auto tensors = EvaluateNodes(output, fetch); in TEST_F() local
731 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
779 auto tensors = EvaluateNodes(output, fetch); in TEST_F() local
930 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
1013 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
1058 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
1098 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
[all …]
Dpin_to_host_optimizer_test.cc66 auto tensors = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
106 auto tensors = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
146 auto tensors = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
182 auto tensors = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
253 auto tensors = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
Dauto_mixed_precision_test.cc201 auto tensors = EvaluateNodes(output, item.fetch, feed); in TestSimpleUnaryInferOp() local
242 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
280 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
329 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
368 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
413 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
454 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
498 auto tensors = EvaluateNodes(output, item.fetch, feed); in TEST_F() local
557 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
597 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
[all …]
Dremapper_test.cc68 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
107 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
212 auto tensors = EvaluateNodes(output, item.fetch, item.feed); in TEST_F() local
324 auto tensors = EvaluateNodes(output, item.fetch, item.feed); in TEST_F() local
388 auto tensors = EvaluateNodes(output, item.fetch, item.feed); in TEST_F() local
447 auto tensors = EvaluateNodes(output, item.fetch, item.feed); in RunTest() local
520 auto tensors = EvaluateNodes(output, item.fetch, item.feed); in RunTest() local
606 auto tensors = EvaluateNodes(output, item.fetch, item.feed); in TEST_F() local
696 auto tensors = EvaluateNodes(output, item.fetch, item.feed); in TEST_F() local
798 auto tensors = EvaluateNodes(output, item.fetch, item.feed); in RunTest() local
[all …]
Dconstant_folding_test.cc121 auto tensors = EvaluateNodes(output, item.fetch, {{"x", x_t}}); in SimpleNeutralElementTest() local
272 auto tensors = in PaddingWithZeroSize() local
305 auto tensors = EvaluateNodes(output, fetch); in TEST_F() local
396 auto tensors = EvaluateNodes(output, fetch, {{"x", x_t}, {"y", y_t}}); in TEST_F() local
451 auto tensors = EvaluateNodes(output, fetch, {{"x", x_t}}); in TEST_F() local
522 auto tensors = EvaluateNodes(output, fetch, {{"x", x_t}}); in TEST_F() local
598 auto tensors = in TEST_F() local
980 auto tensors = EvaluateNodes( in TEST_F() local
1056 auto tensors = EvaluateNodes(output, fetch); in TEST_F() local
1131 auto tensors = EvaluateNodes(output, fetch, in TEST_F() local
[all …]
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dxnnpack_delegate.cc150 std::vector<int> tensors(context->tensors_size, -1); in Create() local
1539 TfLiteNode* node, const TfLiteTensor* tensors, in VisitAbsNode()
1572 TfLiteNode* node, const TfLiteTensor* tensors, in VisitAddNode()
1622 TfLiteNode* node, const TfLiteTensor* tensors, in VisitAveragePool2DNode()
1688 TfLiteNode* node, const TfLiteTensor* tensors, in VisitCeilNode()
1721 TfLiteNode* node, const TfLiteTensor* tensors, in VisitConv2DNode()
1828 TfLiteNode* node, const TfLiteTensor* tensors, in VisitDepthwiseConv2DNode()
1939 TfLiteNode* node, const TfLiteTensor* tensors, in VisitDepthToSpaceNode()
1985 TfLiteNode* node, const TfLiteTensor* tensors, in VisitDivNode()
2035 TfLiteNode* node, const TfLiteTensor* tensors, in VisitEluNode()
[all …]
/external/tensorflow/tensorflow/python/distribute/v1/
Dall_reduce.py31 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/core/common_runtime/
Dall_to_all_test.cc38 std::vector<Tensor> tensors = { in TEST_F() local
70 std::vector<Tensor> tensors = { in TEST_F() local
104 std::vector<Tensor> tensors = { in TEST_F() local
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dquantization_util_test.cc50 void PopulateContext(std::vector<TfLiteTensor>& tensors, in PopulateContext()
160 std::vector<TfLiteTensor> tensors{input, dequantized_input}; in TEST() local
185 std::vector<TfLiteTensor> tensors{input, dequantized_input}; in TEST() local
208 std::vector<TfLiteTensor> tensors{output, quantized_output}; in TEST() local
230 std::vector<TfLiteTensor> tensors{output, quantized_output}; in TEST() local
/external/tensorflow/tensorflow/core/data/
Dsnapshot_utils_test.cc32 std::vector<Tensor>& tensors) { in GenerateTensorVector()
43 std::vector<Tensor> tensors; in SnapshotRoundTrip() local
97 std::vector<Tensor> tensors; in SnapshotReaderBenchmarkLoop() local
153 std::vector<Tensor> tensors; in SnapshotWriterBenchmarkLoop() local
/external/tensorflow/tensorflow/core/kernels/
Dtensor_list.h126 std::vector<Tensor>& tensors() { return tensors_->values_; } in tensors() function
127 const std::vector<Tensor>& tensors() const { return tensors_->values_; } in tensors() function
Dtensor_map.h106 absl::flat_hash_map<TensorKey, Tensor>& tensors() { in tensors() function
110 const absl::flat_hash_map<TensorKey, Tensor>& tensors() const { in tensors() function
/external/tensorflow/tensorflow/python/keras/engine/
Dbase_layer_utils.py158 def have_all_keras_metadata(tensors): argument
166 def create_keras_history(tensors): argument
202 def _create_keras_history_helper(tensors, processed_ops, created_layers): argument
304 def needs_keras_history(tensors, ignore_call_context=False): argument
361 def uses_keras_history(tensors): argument
406 def mark_checked(tensors): argument
/external/tensorflow/tensorflow/cc/framework/
Dtestutil.cc27 void GetTensors(const Scope& scope, OutputList tensors, in GetTensors()
40 const OutputList& tensors, std::vector<Tensor>* out) { in GetTensors()
/external/ComputeLibrary/src/runtime/OMP/
DOMPScheduler.cpp53 ITensorPack tensors; in schedule() local
57 void OMPScheduler::schedule_op(ICPPKernel *kernel, const Hints &hints, ITensorPack &tensors) in schedule_op()
/external/tensorflow/tensorflow/python/distribute/parallel_device/
Dparallel_device.py88 def _pack_tensor(self, *tensors):
106 def pack(self, tensors): argument
/external/tensorflow/tensorflow/core/framework/
Dvariant_tensor_data.cc29 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

1234567891011