| /external/tensorflow/tensorflow/python/distribute/ |
| 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 …]
|
| D | cross_device_utils.py | 320 def build_collective_reduce(input_tensors, argument
377 def build_collective_gather(input_tensors, num_workers, collective_keys): 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 | 219 def get_input_shapes_map(input_tensors): argument
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | models.py | 131 def _clone_functional_model(model, input_tensors=None, layer_fn=_clone_layer): argument
278 def _clone_sequential_model(model, input_tensors=None, layer_fn=_clone_layer): argument
384 def clone_model(model, input_tensors=None, clone_function=None): argument
589 model, input_tensors=None, target_tensors=None, custom_objects=None, argument
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | tape.py | 144 def record_operation(op_type, output_tensors, input_tensors, backward_function, argument
152 def record_operation_backprop_only(op_type, output_tensors, input_tensors, argument
160 def record_operation_forwardprop_only(op_type, output_tensors, input_tensors, argument
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | convert.py | 239 def build_toco_convert_protos(input_tensors, argument
463 def toco_convert_impl(input_data, input_tensors, output_tensors, argument
502 def toco_convert(input_data, input_tensors, output_tensors, *args, **kwargs): argument
|
| D | lite.py | 633 input_tensors, argument
689 def from_session(cls, sess, input_tensors, output_tensors): argument
1155 def from_session(cls, sess, input_tensors, output_tensors): argument
|
| D | util.py | 232 def freeze_graph(sess, input_tensors, output_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 | 142 std::vector<Tensor> input_tensors; in GetNextInternal() local
|
| /external/tensorflow/tensorflow/lite/ |
| D | graph_info.h | 70 std::vector<int> input_tensors; member
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | node.py | 71 input_tensors, argument
|
| D | base_layer_utils.py | 145 def collect_previous_mask(input_tensors): argument
262 def unnest_if_single_tensor(input_tensors): argument
|
| /external/tensorflow/tensorflow/lite/experimental/delegates/hexagon/ |
| D | hexagon_delegate_kernel.cc | 138 std::vector<hexagon_nn_tensordef> input_tensors; in Invoke() local
243 TfLiteContext* context, const TfLiteIntArray* input_tensors, in BuildGraph()
|
| /external/tensorflow/tensorflow/core/profiler/utils/ |
| D | cost_utils.cc | 76 std::vector<std::string> input_tensors; in Predict() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference_testutil.h | 38 std::vector<const Tensor*> input_tensors; member
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_feed.py | 364 def set_configuration_from_input_tensors(self, input_tensors): argument
384 def set_configuration_from_sharded_input_tensors(self, input_tensors): argument
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | operator_property.h | 29 std::vector<int> input_tensors = {}; member
|
| /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
|
| /external/tensorflow/tensorflow/lite/experimental/delegates/hexagon/builders/ |
| D | op_builder.cc | 130 const TfLiteIntArray* input_tensors, TfLiteContext* context) { in AddInputTensors()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_rewriter_transform.cc | 116 std::vector<std::pair<string, Tensor>> input_tensors; in PlaceShapeType() local
|
| /external/tensorflow/tensorflow/python/ |
| D | tfe_wrapper.cc | 109 const py::handle& input_tensors) { in InputTFE_InputTensorHandles()
633 const py::handle& input_tensors, const py::handle& backward_function) { in PYBIND11_MODULE()
641 const py::handle& forwardprop_output_indices) { in PYBIND11_MODULE()
|