Home
last modified time | relevance | path

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

1234

/external/tensorflow/tensorflow/contrib/nccl/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
211 def _apply_all_reduce(reduction, tensors): argument
233 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
185 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/contrib/all_reduce/python/
Dall_reduce.py30 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/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.cc33 const Tensor& VariantTensorData::tensors(int index) const { in tensors() function in tensorflow::VariantTensorData
37 const std::vector<Tensor>& VariantTensorData::tensors() const { in tensors() function in tensorflow::VariantTensorData
Dtensor.proto93 repeated TensorProto tensors = 3; field
/external/tensorflow/tensorflow/contrib/kfac/python/ops/
Destimator.py284 def _get_grads_lists_gradients(self, tensors): argument
292 def _get_grads_lists_empirical(self, tensors): argument
308 def _get_grads_lists_curvature_prop(self, tensors): argument
319 def _get_grads_lists_exact(self, 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.py103 def tensors_compatible(tensors, signatures): argument
134 def create_signatures(tensors): argument
/external/tensorflow/tensorflow/contrib/eager/python/examples/resnet50/
Dresnet50_test.py146 def __init__(self, tensors): argument
245 def make_iterator(tensors): argument
255 def make_iterator(tensors): argument
/external/tensorflow/tensorflow/python/framework/
Dsubscribe.py31 def _recursive_apply(tensors, apply_fn): argument
309 def subscribe(tensors, side_effects): argument
/external/tensorflow/tensorflow/contrib/training/python/training/
Dsampling_ops.py37 def rejection_sample(tensors, argument
134 def stratified_sample(tensors, argument
Dbucket_ops.py63 def bucket(tensors, argument
304 tensors, argument
/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/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/lite/toco/tflite/
Dimport_test.cc65 auto tensors = builder_.CreateVector( in BuildTestModel() local
99 details::TensorsTable tensors; in TEST_F() local
Dimport.cc31 auto tensors = (*input_model.subgraphs())[0]->tensors(); in LoadTensorsTable() local
54 auto tensors = (*input_model.subgraphs())[0]->tensors(); in ImportTensors() local
Dexport_test.cc58 details::TensorsMap tensors; in TEST_F() local
/external/tensorflow/tensorflow/python/training/
Dinput.py419 def _as_tensor_list(tensors): argument
735 def _batch(tensors, batch_size, keep_input, num_threads=1, capacity=32, argument
808 def _shuffle_batch(tensors, batch_size, capacity, min_after_dequeue, argument
898 def batch(tensors, batch_size, num_threads=1, capacity=32, argument
993 def maybe_batch(tensors, keep_input, batch_size, num_threads=1, capacity=32, argument
1205 def shuffle_batch(tensors, batch_size, capacity, min_after_dequeue, argument
1305 def maybe_shuffle_batch(tensors, batch_size, capacity, min_after_dequeue, argument
/external/tensorflow/tensorflow/contrib/slim/python/slim/data/
Dprefetch_queue.py33 def prefetch_queue(tensors, argument
/external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/
Dtest_util.py29 def eval(self, tensors): argument
/external/tensorflow/tensorflow/contrib/tensorboard/plugins/trace/
Dtrace.py136 def _add_data_from_tensors(tensors, info): argument
/external/tensorflow/tensorflow/python/eager/
Dtape.py74 def should_record(tensors): argument

1234