Home
last modified time | relevance | path

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

123456789

/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/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/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/tensorflow/tensorflow/lite/micro/kernels/
Dconv_test_common.cc22 TfLiteStatus InvokeConv(TfLiteTensor* tensors, int tensors_size, in InvokeConv()
42 TfLiteStatus ValidateConvGoldens(TfLiteTensor* tensors, int tensors_size, in ValidateConvGoldens()
60 TfLiteStatus InvokeConv(TfLiteTensor* tensors, int tensors_size, in InvokeConv()
67 TfLiteStatus InvokeConv(TfLiteTensor* tensors, int tensors_size, in InvokeConv()
74 TfLiteStatus InvokeConv(TfLiteTensor* tensors, int tensors_size, in InvokeConv()
81 TfLiteStatus ValidateConvGoldens(TfLiteTensor* tensors, int tensors_size, in ValidateConvGoldens()
92 TfLiteStatus ValidateConvGoldens(TfLiteTensor* tensors, int tensors_size, in ValidateConvGoldens()
103 TfLiteStatus ValidateConvGoldens(TfLiteTensor* tensors, int tensors_size, in ValidateConvGoldens()
130 TfLiteTensor tensors[tensors_size] = { in TestConvFloat() local
163 TfLiteTensor tensors[tensors_size] = { in TestConvQuantizedPerLayer() local
[all …]
Dtranspose_conv_test.cc59 TfLiteStatus InvokeTransposeConv(TfLiteTensor* tensors, int tensors_size, in InvokeTransposeConv()
81 TfLiteStatus ValidateTransposeConvGoldens(TfLiteTensor* tensors, in ValidateTransposeConvGoldens()
118 TfLiteTensor tensors[tensors_size] = { in TestTransposeConvFloat() local
163 TfLiteTensor tensors[tensors_size] = { in TestTransposeConvQuantized() local
244 TfLiteTensor tensors[tensors_size] = { in TF_LITE_MICRO_TEST() local
280 TfLiteTensor tensors[tensors_size] = { in TF_LITE_MICRO_TEST() 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.cc148 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
180 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
226 auto tensors = EvaluateNodes(output, fetch); in TEST_F() local
259 auto tensors = EvaluateNodes(output, fetch); in TEST_F() local
305 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
353 auto tensors = EvaluateNodes(output, fetch); in TEST_F() local
504 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
587 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
632 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
672 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.cc197 auto tensors = EvaluateNodes(output, item.fetch, feed); in TestSimpleUnaryInferOp() local
238 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
276 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
325 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
364 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
409 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
450 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
494 auto tensors = EvaluateNodes(output, item.fetch, feed); in TEST_F() local
553 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
593 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
452 auto tensors = EvaluateNodes(output, item.fetch, item.feed); in RunTest() local
538 auto tensors = EvaluateNodes(output, item.fetch, item.feed); in TEST_F() local
628 auto tensors = EvaluateNodes(output, item.fetch, item.feed); in TEST_F() local
730 auto tensors = EvaluateNodes(output, item.fetch, item.feed); in RunTest() local
826 auto tensors = EvaluateNodes(output, item.fetch, item.feed); in TEST_F() 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
975 auto tensors = EvaluateNodes( in TEST_F() local
1051 auto tensors = EvaluateNodes(output, fetch); in TEST_F() local
1126 auto tensors = EvaluateNodes(output, fetch, in TEST_F() local
[all …]
Dcommon_subgraph_elimination_test.cc94 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
133 auto tensors = EvaluateNodes(output, item.fetch, {{"Placeholder", bool_t}}); in TEST_F() local
172 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dxnnpack_delegate.cc143 std::vector<int> tensors(context->tensors_size, -1); in Create() local
1042 TfLiteNode* node, const TfLiteTensor* tensors, in VisitAbsNode()
1075 TfLiteNode* node, const TfLiteTensor* tensors, in VisitAddNode()
1125 TfLiteNode* node, const TfLiteTensor* tensors, in VisitAveragePool2DNode()
1191 TfLiteNode* node, const TfLiteTensor* tensors, in VisitCeilNode()
1224 TfLiteNode* node, const TfLiteTensor* tensors, in VisitConv2DNode()
1322 TfLiteNode* node, const TfLiteTensor* tensors, in VisitDepthwiseConv2DNode()
1423 TfLiteNode* node, const TfLiteTensor* tensors, in VisitDepthToSpaceNode()
1469 TfLiteNode* node, const TfLiteTensor* tensors, in VisitDivNode()
1519 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/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/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/kernels/data/experimental/
Dsnapshot_util_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/python/keras/engine/
Dbase_layer_utils.py161 def have_all_keras_metadata(tensors): argument
169 def create_keras_history(tensors): argument
205 def _create_keras_history_helper(tensors, processed_ops, created_layers): argument
307 def needs_keras_history(tensors, ignore_call_context=False): argument
364 def uses_keras_history(tensors): argument
409 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/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
/external/tensorflow/tensorflow/c/eager/parallel_device/
Dparallel_device_lib.h187 std::vector<TensorHandlePtr> tensors, in ParallelTensor()
194 std::vector<TensorHandlePtr> tensors, const TF_DataType dtype) in ParallelTensor()
/external/tensorflow/tensorflow/python/framework/
Dsubscribe.py30 def _recursive_apply(tensors, apply_fn): argument
313 def subscribe(tensors, side_effects): argument
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_memory_test.cc58 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

123456789