| /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/lite/experimental/mlir/testing/ |
| D | mlir_convert.py | 30 def mlir_convert(options, graph_def, input_tensors, output_tensors, **kwargs): argument
76 def representative_dataset(input_tensors): argument
126 input_tensors, argument
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | models.py | 129 def _clone_functional_model(model, input_tensors=None, layer_fn=_clone_layer): argument
279 def _clone_sequential_model(model, input_tensors=None, layer_fn=_clone_layer): argument
385 def clone_model(model, input_tensors=None, clone_function=None): argument
618 model, input_tensors=None, target_tensors=None, custom_objects=None, argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/benchmarks/ |
| D | benchmark_mlir_function.cc | 37 llvm::SmallVector<Tensor> input_tensors; in GetInputTensors() local
77 llvm::SmallVector<Tensor> input_tensors = GetInputTensors(input_specs); in RunMlirBenchmark() local
133 llvm::SmallVector<Tensor> input_tensors = GetInputTensors(input_specs); in RunEigenBenchmark() local
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | toco_convert.py | 76 def toco_convert(options, graph_def, input_tensors, output_tensors, **kwargs): argument
122 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
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | lite.py | 819 def _validate_inputs(self, graph_def, input_tensors): argument
863 def _optimize_tf_model(self, graph_def, input_tensors, output_tensors, argument
889 def convert(self, graph_def, input_tensors, output_tensors): argument
1594 def _validate_inputs(self, input_tensors, quantized_input_stats): argument
1643 def _optimize_tf_model(self, graph_def, input_tensors, output_tensors, argument
2028 input_tensors, argument
2207 input_tensors, argument
2239 def from_session(cls, sess, input_tensors, output_tensors): argument
2453 def from_session(cls, sess, input_tensors, output_tensors): argument
|
| D | convert.py | 496 def build_toco_convert_protos(input_tensors, argument
774 def toco_convert_impl(input_data, input_tensors, output_tensors, argument
839 def toco_convert(input_data, input_tensors, output_tensors, *args, **kwargs): argument
|
| /external/tensorflow/tensorflow/lite/ |
| D | graph_info.h | 78 std::vector<int> input_tensors; member
|
| /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 | 228 def input_tensors(self): member in Node
|
| D | base_layer_utils.py | 142 def collect_previous_mask(input_tensors): argument
293 def unnest_if_single_tensor(input_tensors): argument
|
| /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/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/lite/tools/optimize/ |
| D | operator_property.h | 50 std::vector<int> input_tensors = {}; member
|
| /external/tensorflow/tensorflow/lite/delegates/ |
| D | serialization.cc | 273 auto* input_tensors = delegate_params->input_tensors; in GetEntryImpl() local
|