Home
last modified time | relevance | path

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

12345

/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.py32 def _DeviceTensors(tensors, devices): argument
40 def _NcclAllReduce(nccl_fun, tensors, devices): argument
44 def _NcclReduce(nccl_fun, tensors, devices): argument
50 def _NcclBroadcast(tensors, devices): argument
119 def _Gradient(tensors, devices): argument
186 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/contrib/slim/python/slim/
Dsummaries.py152 def add_histogram_summaries(tensors, prefix=None): argument
169 def add_image_summaries(tensors, prefix=None): argument
186 def add_scalar_summaries(tensors, prefix=None, print_summary=False): argument
206 def add_zero_fraction_summaries(tensors, prefix=None): argument
/external/tensorflow/tensorflow/lite/toco/tflite/
Dimport_test.cc104 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/python/keras/engine/
Dbase_layer_utils.py228 def have_all_keras_metadata(tensors): argument
236 def create_keras_history(tensors): argument
254 def _create_keras_history_helper(tensors, processed_ops=None): argument
300 def needs_keras_history(tensors): argument
328 def uses_keras_history(tensors): argument
369 def mark_checked(tensors): argument
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_optimizer_test.cc114 auto tensors = EvaluateFetchNodes(optimized); in TEST_F() local
190 auto tensors = EvaluateFetchNodes(optimized); in TEST_F() local
290 auto tensors = EvaluateFetchNodes(optimized); in TEST_F() local
374 auto tensors = EvaluateFetchNodes(optimized); in TEST_F() local
424 auto tensors = EvaluateFetchNodes(optimized); in TEST_F() local
555 auto tensors = EvaluateFetchNodes(optimized); in TEST_F() local
764 auto tensors = EvaluateFetchNodes(optimized); in TEST_F() local
906 auto tensors = EvaluateFetchNodes(optimized); in TEST_F() local
1065 auto tensors = EvaluateFetchNodes(optimized); in TEST_F() local
1135 auto tensors = EvaluateFetchNodes(item); in TEST_F() local
[all …]
Darithmetic_optimizer_test.cc129 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
169 auto tensors = EvaluateNodes(output, item.fetch, {{"Placeholder", bool_t}}); in TEST_F() local
208 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
242 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
273 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
317 auto tensors = EvaluateNodes(output, fetch); in TEST_F() local
350 auto tensors = EvaluateNodes(output, fetch); in TEST_F() local
393 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
438 auto tensors = EvaluateNodes(output, fetch); in TEST_F() local
585 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
[all …]
Dremapper_test.cc63 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
98 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
161 auto tensors = EvaluateNodes(output, item.fetch, item.feed); in TEST_F() local
226 auto tensors = EvaluateNodes(output, item.fetch, item.feed); in TEST_F() local
303 auto tensors = EvaluateNodes(output, item.fetch, item.feed); in TEST_F() local
382 auto tensors = EvaluateNodes(output, item.fetch, item.feed); in TEST_F() local
454 auto tensors = EvaluateNodes(output, item.fetch, item.feed); in TEST_F() local
Dpin_to_host_optimizer_test.cc86 auto tensors = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
129 auto tensors = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
165 auto tensors = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
236 auto tensors = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
Dconstant_folding_test.cc115 auto tensors = EvaluateNodes(output, item.fetch, {{"x", x_t}}); in SimpleNeutralElementTest() local
245 auto tensors = EvaluateNodes(output, fetch); in TEST_F() local
333 auto tensors = EvaluateNodes(output, fetch); in TEST_F() local
661 auto tensors = EvaluateNodes( in TEST_F() local
737 auto tensors = EvaluateNodes(output, fetch); in TEST_F() local
812 auto tensors = EvaluateNodes(output, fetch, in TEST_F() local
880 auto tensors = EvaluateNodes(output, fetch, in TEST_F() local
979 auto tensors = EvaluateNodes(output, fetch); in TEST_F() local
1122 auto tensors = EvaluateNodes(output, item.fetch); in TEST_F() local
1249 auto tensors = EvaluateNodes(output, item.fetch, in TEST_F() local
[all …]
/external/tensorflow/tensorflow/core/framework/
Dunique_tensor_references_test.cc32 TensorReferenceVector tensors; in TEST() local
55 TensorReferenceVector tensors; in TEST() local
91 TensorReferenceVector tensors; in TEST() local
116 TensorReferenceVector tensors; in TEST() local
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/python/distribute/
Dall_reduce.py31 def _flatten_tensors(tensors): argument
60 def _reshape_tensors(tensors, shape): argument
131 def _strip_padding(tensors, pad_len): argument
842 def upper_builder(tensors): argument
845 def upper_level_f(tensors): argument
854 def upper_builder(tensors): argument
857 def upper_level_f(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/contrib/learn/python/learn/estimators/
Dtensor_signature.py112 def tensors_compatible(tensors, signatures): argument
143 def create_signatures(tensors): argument
/external/tensorflow/tensorflow/contrib/training/python/training/
Dsampling_ops.py38 def rejection_sample(tensors, argument
135 def stratified_sample(tensors, argument
Dbucket_ops.py63 def bucket(tensors, argument
304 tensors, argument
/external/tensorflow/tensorflow/python/framework/
Dsubscribe.py31 def _recursive_apply(tensors, apply_fn): argument
304 def subscribe(tensors, side_effects): argument
Dconvert_to_constants.py62 def _get_tensors_from_graph(graph, tensors): 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
/external/tensorflow/tensorflow/contrib/opt/python/training/
Dexternal_optimizer.py251 def _pack(cls, tensors): argument
261 def _make_eval_func(self, tensors, session, feed_dict, fetches, argument
288 tensors, argument
/external/tensorflow/tensorflow/contrib/slim/python/slim/data/
Dprefetch_queue.py34 def prefetch_queue(tensors, argument
/external/tensorflow/tensorflow/contrib/eager/python/examples/densenet/
Ddensenet_test.py176 def __init__(self, tensors): argument
327 def make_iterator(tensors): argument
338 def make_iterator(tensors): argument
/external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/
Dtest_util.py29 def eval(self, tensors): argument

12345