| /external/tensorflow/tensorflow/python/distribute/v1/ |
| D | all_reduce.py | 250 def build_ring_all_reduce(input_tensors, num_workers, num_subchunks, argument
293 def _build_ring_gather(input_tensors, devices, num_subchunks, argument
422 def build_recursive_hd_all_reduce(input_tensors, red_op, un_op=None): argument
476 def _build_recursive_hd_gather(input_tensors, devices, red_op): argument
517 def _build_recursive_hd_scatter(input_tensors, devices): argument
554 def build_shuffle_all_reduce(input_tensors, gather_devices, red_op, un_op=None): argument
588 def _build_shuffle_gather(input_tensors, gather_devices, red_op, un_op=None): argument
681 def build_nccl_all_reduce(input_tensors, red_op, un_op=None): argument
710 def _build_nccl_hybrid(input_tensors, red_op, upper_level_f): argument
761 def _reduce_non_singleton(input_tensors, red_f, un_op): argument
[all …]
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | mlir_convert.py | 31 input_tensors, argument
84 def representative_dataset(input_tensors): argument
133 input_tensors, argument
|
| /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/compiler/mlir/tfrt/benchmarks/ |
| D | benchmark_mlir_function.cc | 52 llvm::SmallVector<Tensor> input_tensors; in GetInputTensors() local
105 llvm::SmallVector<Tensor> input_tensors = GetInputTensors(input_specs); in RunJitRtBenchmark() local
179 llvm::SmallVector<Tensor> input_tensors = GetInputTensors(input_specs); in RunTfrtBenchmark() local
205 llvm::SmallVector<Tensor> input_tensors = GetInputTensors(input_specs); in RunEigenBenchmark() local
|
| /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/python/eager/ |
| D | tape.py | 177 def record_operation(op_type, output_tensors, input_tensors, backward_function, argument
185 def record_operation_backprop_only(op_type, output_tensors, input_tensors, argument
193 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
245 auto input_tensors = test_case.map_defun_op_params.GetInputTensors(); in TEST_P() local
274 auto input_tensors = test_case.map_defun_op_params.GetInputTensors(); in TEST_F() local
|
| D | zip_dataset_op.cc | 137 std::vector<Tensor> input_tensors; in Get() local
194 std::vector<Tensor> input_tensors; in GetNextInternal() local
|
| D | reduce_dataset_op_test.cc | 51 std::vector<Tensor> input_tensors = initial_state_; in GetInputTensors() local
|
| /external/pytorch/test/cpp/aoti_inference/ |
| D | test.cpp | 32 auto input_tensors = in test_aoti() local
99 auto input_tensors = in test_aoti_constants_update() local
188 auto input_tensors = in test_aoti_double_buffering() local
265 auto input_tensors = in test_aoti_double_buffering_with_tensor_constants() local
|
| /external/pytorch/torch/fx/experimental/shape_inference/ |
| D | infer_shape.py | 21 def infer_shape(gm, input_tensors): argument
|
| /external/tflite-support/tensorflow_lite_support/cc/task/text/nlclassifier/ |
| D | bert_nl_classifier.cc | 68 const std::vector<TfLiteTensor*>& input_tensors, const std::string& input) { in Preprocess()
205 const auto& input_tensors = GetInputTensors(); in InitializeFromMetadata() local
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | lite.py | 1000 def _validate_inputs(self, graph_def, input_tensors): argument
1047 def _optimize_tf_model(self, graph_def, input_tensors, output_tensors, argument
1103 def convert(self, graph_def, input_tensors, output_tensors): argument
1945 def _validate_inputs(self, input_tensors, quantized_input_stats): argument
1994 def _optimize_tf_model(self, graph_def, input_tensors, output_tensors, argument
2379 input_tensors, argument
2560 input_tensors, argument
2592 def from_session(cls, sess, input_tensors, output_tensors): argument
2819 def from_session(cls, sess, input_tensors, output_tensors): argument
|
| D | convert.py | 720 def convert_graphdef(input_data, input_tensors, output_tensors, **kwargs): argument
844 def toco_convert(input_data, input_tensors, output_tensors, *args, **kwargs): argument
|
| /external/tensorflow/tensorflow/distribute/experimental/rpc/proto/ |
| D | tf_rpc_service.proto | 10 repeated TensorProto input_tensors = 2; field
|
| /external/tensorflow/tensorflow/lite/ |
| D | graph_info.h | 80 std::vector<int> input_tensors; member
|
| /external/pytorch/benchmarks/operator_benchmark/common/ |
| D | repeat_benchmark.py | 42 input_tensors, total_bytes = generate_data_for_repeat() variable
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | split_utils_test.cc | 71 const std::vector<Tensor>& input_tensors, in RunFunction()
107 const std::vector<Tensor>& input_tensors, in RunSplitFunctions()
155 const std::vector<Tensor>& input_tensors, in CheckSplitFunctions()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/ |
| D | hexagon_delegate_kernel.cc | 98 std::vector<hexagon_nn_tensordef> input_tensors; in Eval() local
244 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/saved_model/model_utils/ |
| D | export_utils.py | 67 def _ensure_servable(input_tensors, names_to_output_tensor_infos): argument
|
| /external/tensorflow/tensorflow/core/transforms/utils/ |
| D | eval_utils.cc | 98 absl::InlinedVector<tensorflow::Tensor, 4> input_tensors(operands.size()); in EvaluateOperation() local
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_feed.py | 376 def set_configuration_from_input_tensors(self, input_tensors): argument
396 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
|