| /external/tensorflow/tensorflow/python/distribute/v1/ |
| D | all_reduce.py | 254 def build_ring_all_reduce(input_tensors, num_workers, num_subchunks, argument
297 def _build_ring_gather(input_tensors, devices, num_subchunks, argument
426 def build_recursive_hd_all_reduce(input_tensors, red_op, un_op=None): argument
480 def _build_recursive_hd_gather(input_tensors, devices, red_op): argument
521 def _build_recursive_hd_scatter(input_tensors, devices): argument
558 def build_shuffle_all_reduce(input_tensors, gather_devices, red_op, un_op=None): argument
592 def _build_shuffle_gather(input_tensors, gather_devices, red_op, un_op=None): argument
685 def build_nccl_all_reduce(input_tensors, red_op, un_op=None): argument
714 def _build_nccl_hybrid(input_tensors, red_op, upper_level_f): argument
765 def _reduce_non_singleton(input_tensors, red_f, un_op): argument
[all …]
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | models.py | 133 def _clone_functional_model(model, input_tensors=None, layer_fn=_clone_layer): argument
283 def _clone_sequential_model(model, input_tensors=None, layer_fn=_clone_layer): argument
389 def clone_model(model, input_tensors=None, clone_function=None): argument
597 model, input_tensors=None, target_tensors=None, custom_objects=None, argument
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | toco_convert.py | 78 def toco_convert(options, graph_def, input_tensors, output_tensors, **kwargs): argument
125 def representative_dataset(input_tensors): argument
|
| D | zip_test_utils.py | 230 def get_input_shapes_map(input_tensors): argument
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | tape.py | 181 def record_operation(op_type, output_tensors, input_tensors, backward_function, argument
189 def record_operation_backprop_only(op_type, output_tensors, input_tensors, argument
197 def record_operation_forwardprop_only(op_type, output_tensors, input_tensors, argument
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | map_defun_op_test.cc | 44 std::vector<Tensor> input_tensors = arguments_; in GetInputTensors() local
244 auto input_tensors = test_case.map_defun_op_params.GetInputTensors(); in TEST_P() local
273 auto input_tensors = test_case.map_defun_op_params.GetInputTensors(); in TEST_F() local
|
| D | reduce_dataset_op_test.cc | 51 std::vector<Tensor> input_tensors = initial_state_; in GetInputTensors() local
|
| D | zip_dataset_op.cc | 150 std::vector<Tensor> input_tensors; in GetNextInternal() local
|
| /external/tensorflow/tensorflow/lite/ |
| D | graph_info.h | 78 std::vector<int> input_tensors; member
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | convert.py | 363 def build_toco_convert_protos(input_tensors, argument
599 def toco_convert_impl(input_data, input_tensors, output_tensors, argument
662 def toco_convert(input_data, input_tensors, output_tensors, *args, **kwargs): argument
|
| D | lite.py | 607 def convert(self, graph_def, input_tensors, output_tensors): argument
1630 input_tensors, argument
1790 input_tensors, argument
1822 def from_session(cls, sess, input_tensors, output_tensors): argument
2036 def from_session(cls, sess, input_tensors, output_tensors): argument
|
| D | util.py | 303 def freeze_graph(sess, input_tensors, output_tensors): argument
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/ |
| D | hexagon_delegate_kernel.cc | 96 std::vector<hexagon_nn_tensordef> input_tensors; in Eval() local
240 TfLiteContext* context, const TfLiteIntArray* input_tensors, in BuildGraph()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference_testutil.h | 39 std::vector<const Tensor*> input_tensors; member
|
| /external/tflite-support/tensorflow_lite_support/cc/task/vision/core/ |
| D | base_vision_task_api.h | 114 absl::Status Preprocess(const std::vector<TfLiteTensor*>& input_tensors, in Preprocess()
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_feed.py | 377 def set_configuration_from_input_tensors(self, input_tensors): argument
397 def set_configuration_from_sharded_input_tensors(self, input_tensors): argument
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | node.py | 231 def input_tensors(self): member in Node
|
| D | base_layer_utils.py | 145 def collect_previous_mask(input_tensors): argument
296 def unnest_if_single_tensor(input_tensors): argument
|
| /external/tflite-support/tensorflow_lite_support/cc/task/text/nlclassifier/ |
| D | bert_nl_classifier.cc | 65 const std::vector<TfLiteTensor*>& input_tensors, const std::string& input) { in Preprocess()
|
| D | nl_classifier.cc | 188 const std::vector<TfLiteTensor*>& input_tensors, const std::string& input) { in Preprocess()
|
| /external/tensorflow/tensorflow/tools/benchmark/ |
| D | benchmark_model.cc | 75 void CreateTensorsFromInputInfo( in CreateTensorsFromInputInfo()
120 std::vector<std::pair<string, tensorflow::Tensor> > input_tensors; in GetOutputShapes() local
288 std::vector<std::pair<string, tensorflow::Tensor> > input_tensors; in RunBenchmark() local
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_function.cc | 52 const TF_Output* inputs, std::vector<OutputTensor>* input_tensors, in ProcessInputs()
161 std::vector<tensorflow::OutputTensor> input_tensors; in TF_GraphToFunctionWithControlOutputs() local
|
| D | c_api_experimental_test.cc | 106 const std::vector<TF_Tensor*>& input_tensors, in CheckOutputShapes()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | operator_property.h | 50 std::vector<int> input_tensors = {}; member
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_rewriter_transform.cc | 116 std::vector<std::pair<string, Tensor>> input_tensors; in PlaceShapeType() local
|