Searched defs:data_inputs (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| H A D | tpu_embedding_software_deduplication_rewrite_pass.cc | 103 absl::Span<const NodeDefBuilder::NodeOut> data_inputs, in MakeRecvActivationsNodeDef()
167 absl::Span<const NodeDefBuilder::NodeOut> data_inputs, in MakeSendGradientsNodeDef()
304 std::vector<NodeDefBuilder::NodeOut> data_inputs; in GetDataInputs() local
480 const std::vector<NodeDefBuilder::NodeOut> data_inputs = in ReplaceSendGradientsNodeAndAddDeduplicationInputs() local
715 std::vector<NodeDefBuilder::NodeOut> data_inputs; in GetDataInputs() local
|
| H A D | encapsulate_tpu_computations_pass.cc | 2652 std::vector<std::pair<Node*, int>> data_inputs(num_inputs); in BuildTPUReplicateOps() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | dynamic_stitch_op.cc | 77 OpInputList* data_inputs, int* first_dim_size, in CheckArgsAndAllocateResult()
166 OpInputList data_inputs; in Compute() local
238 OpInputList data_inputs; in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| H A D | directed_interleave_dataset_op.cc | 50 std::vector<DatasetBase*> data_inputs, bool stop_on_empty_dataset) in Dataset()
350 std::vector<DatasetBase*> data_inputs; in MakeDataset() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | accumulate_n_optimizer.cc | 243 std::vector<NodeBuilder::NodeOut> data_inputs; in RewriteIntoAddN() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | encapsulate_xla_computations_pass.cc | 264 std::vector<std::pair<Node*, int>> data_inputs(num_inputs); in BuildXlaLaunchOps() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | functionalize_cond.cc | 1165 int data_inputs = 0; in DetermineCondStateMerge() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | function_optimizer.cc | 1133 std::set<const Node*> data_inputs; in AddStrictInputSemantics() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | function.cc | 499 std::vector<std::pair<int, int>> data_inputs; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/ops/ |
| H A D | dataset_ops.py | 6392 def __init__(self, selector_input, data_inputs, stop_on_empty_dataset=False): argument
|