Searched refs:in_tensors (Results 1 – 17 of 17) sorted by relevance
| /external/tensorflow/tensorflow/lite/python/ |
| D | convert_saved_model_test.py | 93 graph_def, in_tensors, out_tensors, _ = (
101 return graph_def, in_tensors, out_tensors
106 _, in_tensors, out_tensors = self._convertSavedModel(saved_model_dir)
109 self.assertEqual(self._getArrayNames(in_tensors), ["Placeholder:0"])
110 self.assertEqual(self._getArrayShapes(in_tensors), [[1, 16, 16, 3]])
115 _, in_tensors, out_tensors = self._convertSavedModel(saved_model_dir)
118 self.assertEqual(self._getArrayNames(in_tensors), ["Placeholder:0"])
119 self.assertEqual(self._getArrayShapes(in_tensors), [[None, 16, 16, 3]])
158 _, in_tensors, out_tensors = self._convertSavedModel(
164 self.assertEqual(self._getArrayNames(in_tensors), ["Placeholder:0"])
[all …]
|
| D | convert_saved_model.py | 202 in_tensors = _get_tensors(graph, inputs, input_arrays)
204 util.set_tensor_shapes(in_tensors, input_shapes)
206 frozen_graph_def = util.freeze_graph(sess, in_tensors, out_tensors)
207 return frozen_graph_def, in_tensors, out_tensors, sess.graph
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_Batch.pbtxt | 9 All Tensors in in_tensors are batched together (so, for example, labels and
20 in_tensors must have the first dimension of the same size.
22 in_tensors: The tensors to be batched.
|
| D | api_def_BatchFunction.pbtxt | 4 name: "in_tensors"
112 in_tensors=[a],
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | Batch.pbtxt | 4 name: "in_tensors"
74 name: "in_tensors"
|
| D | BatchFunction.pbtxt | 4 name: "in_tensors"
88 name: "in_tensors"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | Batch.pbtxt | 4 name: "in_tensors"
74 name: "in_tensors"
|
| D | BatchFunction.pbtxt | 4 name: "in_tensors"
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | batch_ops.py | 116 in_tensors=list(args),
|
| D | batch_ops_test.py | 399 in_tensors=[inp],
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batch_kernels.cc | 344 OpInputList in_tensors; in InstantiateFunction() local
345 TF_RETURN_IF_ERROR(c->input_list("in_tensors", &in_tensors)); in InstantiateFunction()
346 for (int i = 0; i < in_tensors.size(); i++) { in InstantiateFunction()
347 if (in_tensors[i].dtype() == DT_RESOURCE) { in InstantiateFunction()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops.td | 1327 in_tensors=[a],
1354 Variadic<TF_Tensor>:$in_tensors,
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 317 …argspec: "args=[\'in_tensors\', \'num_batch_threads\', \'max_batch_size\', \'batch_timeout_micros\…
349 …argspec: "args=[\'in_tensors\', \'captured_tensors\', \'f\', \'num_batch_threads\', \'max_batch_si…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 317 …argspec: "args=[\'in_tensors\', \'num_batch_threads\', \'max_batch_size\', \'batch_timeout_micros\…
349 …argspec: "args=[\'in_tensors\', \'captured_tensors\', \'f\', \'num_batch_threads\', \'max_batch_si…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 43701 func Batch(scope *Scope, in_tensors []tf.Output, num_batch_threads int64, max_batch_size int64, bat…
43712 tf.OutputList(in_tensors),
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 3294 name: "in_tensors"
3535 name: "in_tensors"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 8771 name: "in_tensors"
8841 name: "in_tensors"
9098 name: "in_tensors"
|