| /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
474 def _build_recursive_hd_gather(input_tensors, devices, red_op): argument
515 def _build_recursive_hd_scatter(input_tensors, devices): argument
552 def build_shuffle_all_reduce(input_tensors, gather_devices, red_op, un_op=None): argument
586 def _build_shuffle_gather(input_tensors, gather_devices, red_op, un_op=None): argument
679 def build_nccl_all_reduce(input_tensors, red_op, un_op=None): argument
708 def _build_nccl_hybrid(input_tensors, red_op, upper_level_f): argument
759 def _reduce_non_singleton(input_tensors, red_f, un_op): argument
[all …]
|
| D | cross_device_utils.py | 328 def build_collective_reduce(input_tensors, argument
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | models.py | 53 def _clone_functional_model(model, input_tensors=None, share_weights=False): argument
175 def _clone_sequential_model(model, input_tensors=None, share_weights=False): argument
251 def clone_model(model, input_tensors=None): argument
436 model, input_tensors=None, target_tensors=None, custom_objects=None, argument
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | convert.py | 232 def build_toco_convert_protos(input_tensors, argument
425 def toco_convert_impl(input_data, input_tensors, output_tensors, *args, argument
458 def toco_convert(input_data, input_tensors, output_tensors, *args, **kwargs): argument
|
| D | lite.py | 405 input_tensors, argument
457 def from_session(cls, sess, input_tensors, output_tensors): argument
826 def from_session(cls, sess, input_tensors, output_tensors): argument
|
| /external/tensorflow/tensorflow/lite/tools/accuracy/ |
| D | run_tflite_model_op.cc | 31 Status ValidateInputsMatch(const OpInputList& input_tensors, in ValidateInputsMatch()
86 OpInputList input_tensors; in Compute() local
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/utils/ |
| D | saved_model_export_utils.py | 86 def build_standardized_signature_def(input_tensors, output_tensors, argument
150 def _is_classification_problem(problem_type, input_tensors, output_tensors): argument
160 def _is_regression_problem(problem_type, input_tensors, output_tensors): argument
|
| /external/tensorflow/tensorflow/lite/ |
| D | graph_info.h | 66 std::vector<int> input_tensors; member
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_internal.h | 148 std::vector<std::pair<tensorflow::TensorHandle*, TF_Output>>* input_tensors = member
154 delete input_tensors; in ~TFE_TraceContext() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference_testutil.h | 38 std::vector<const Tensor*> input_tensors; member
|
| D | shape_inference.cc | 35 const std::vector<const Tensor*>& input_tensors, in InferenceContext()
90 const std::vector<const Tensor*>& input_tensors, in InferenceContext()
143 const std::vector<const Tensor*>& input_tensors, in InferenceContext()
225 const OpDef& op_def, const std::vector<const Tensor*>& input_tensors, in PreInputInit()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | tape.py | 132 def record_operation(op_type, output_tensors, input_tensors, backward_function): argument
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_feed.py | 359 def set_configuration_from_input_tensors(self, input_tensors): argument
379 def set_configuration_from_sharded_input_tensors(self, input_tensors): argument
|
| /external/tensorflow/tensorflow/tools/benchmark/ |
| D | benchmark_model.cc | 74 void CreateTensorsFromInputInfo( in CreateTensorsFromInputInfo()
119 std::vector<std::pair<string, tensorflow::Tensor> > input_tensors; in GetOutputShapes() local
287 std::vector<std::pair<string, tensorflow::Tensor> > input_tensors; in RunBenchmark() local
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | cpp_shape_inference.cc | 103 std::vector<const Tensor*> input_tensors; in RunCppShapeInferenceImpl() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.cc | 255 std::vector<const Tensor*> input_tensors(node->num_inputs(), nullptr); in AddNode() local
625 std::vector<const Tensor*> input_tensors(node->num_inputs(), nullptr); in RunShapeFn() local
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | zip_dataset_op.cc | 137 std::vector<Tensor> input_tensors; in GetNextInternal() local
|
| /external/tensorflow/tensorflow/contrib/bigtable/kernels/ |
| D | bigtable_lookup_dataset_op.cc | 135 std::vector<Tensor> input_tensors; in GetNextInternal() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_rewriter_transform.cc | 116 std::vector<std::pair<string, Tensor>> input_tensors; in PlaceShapeType() local
|
| D | remote_fused_graph_rewriter_transform_test.cc | 99 const std::vector<std::pair<string, Tensor>> input_tensors{ in FuseWithPlacedArgs() local
|
| D | remote_fused_graph_execute_utils.cc | 208 std::vector<std::pair<string, tensorflow::Tensor>> input_tensors; in DryRunInference() local
578 const std::vector<std::pair<string, Tensor>>& input_tensors, in BuildAndAddTensorShapes()
1174 const std::vector<std::pair<string, Tensor>>& input_tensors, in FuseRemoteGraphByPlacedArguments()
1319 const std::vector<std::pair<string, Tensor>>& input_tensors) { in IsFuseReady()
|
| /external/tensorflow/tensorflow/contrib/tpu/python/tpu/ |
| D | keras_support.py | 534 def make_input_specs(self, input_tensors): argument
622 def make_input_specs(self, input_tensors): argument
717 def make_input_specs(self, input_tensors): argument
879 core_id_place_holder, input_tensors, inputs): argument
|
| /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/ |
| D | head.py | 462 def _identity_metric_nested(name, input_tensors): argument
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | base_layer_utils.py | 212 def collect_previous_mask(input_tensors): argument
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_function.cc | 462 const TF_Output* inputs, std::vector<OutputTensor>* input_tensors, in ProcessInputs()
571 std::vector<tensorflow::OutputTensor> input_tensors; in TF_GraphToFunctionWithControlOutputs() local
|