Searched refs:input_it (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| D | broadcast.cc | 48 auto input_it = input_dims.rbegin(); in BroadcastTo() local
51 if (input_it != input_dims.rend()) { in BroadcastTo()
52 if (!(*output_it == 0 && *input_it == 0) && in BroadcastTo()
53 !(*input_it != 0 && *output_it % *input_it == 0)) { in BroadcastTo()
61 if (*output_it == *input_it) { in BroadcastTo()
63 } else if (*output_it != *input_it) { in BroadcastTo()
67 broadcast_shape.push_back(*input_it); in BroadcastTo()
68 broadcast_shape.push_back(*output_it / *input_it); in BroadcastTo()
70 ++input_it; in BroadcastTo()
75 TF_RET_CHECK(input_it == input_dims.rend()); in BroadcastTo()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_tensorflow_switch.cc | 93 auto input_it = other_op->inputs.begin(); in Run() local
94 while (input_it != other_op->inputs.end()) { in Run()
95 if (*input_it == switch_op->outputs[nonselected_output_index]) { in Run()
101 input_it = other_op->inputs.erase(input_it); in Run()
103 ++input_it; in Run()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | utils.h | 127 auto input_it = is_ctrl ? input_name.begin() + 1 : input_name.begin(); in NodePositionIfSameNode() local
130 std::distance(input_it, input_name.end()) < node_name.size()) { in NodePositionIfSameNode()
134 if (*input_it++ != *node_it++) { in NodePositionIfSameNode()
138 if (input_it == input_name.end()) { in NodePositionIfSameNode()
140 } else if (*input_it++ == ':') { in NodePositionIfSameNode()
141 StringPiece remaining(&(*input_it), in NodePositionIfSameNode()
142 std::distance(input_it, input_name.end())); in NodePositionIfSameNode()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | spectrogram.cc | 201 auto input_it = input.begin() + *input_start; in GetNextWindowOfSamples() local
202 int input_remaining = input.end() - input_it; in GetNextWindowOfSamples()
205 input_queue_.insert(input_queue_.end(), input_it, input.end()); in GetNextWindowOfSamples()
212 input_queue_.insert(input_queue_.end(), input_it, in GetNextWindowOfSamples()
213 input_it + samples_to_next_step_); in GetNextWindowOfSamples()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | spectrogram.cc | 170 auto input_it = input.begin() + *input_start; in GetNextWindowOfSamples() local
171 int input_remaining = input.end() - input_it; in GetNextWindowOfSamples()
174 input_queue_.insert(input_queue_.end(), input_it, input.end()); in GetNextWindowOfSamples()
181 input_queue_.insert(input_queue_.end(), input_it, in GetNextWindowOfSamples()
182 input_it + samples_to_next_step_); in GetNextWindowOfSamples()
|
| /external/v8/src/compiler/ |
| D | branch-elimination.cc | 173 auto input_it = inputs.begin(); in ReduceMerge() local
177 ControlPathConditions conditions = node_conditions_.Get(*input_it); in ReduceMerge()
178 ++input_it; in ReduceMerge()
182 for (; input_it != input_end; ++input_it) { in ReduceMerge()
186 conditions.ResetToCommonAncestor(node_conditions_.Get(*input_it)); in ReduceMerge()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | fusion_utils.cc | 219 auto input_it = in FuseFunctionNodes() local
221 if (input_it == second_inputs.end()) continue; in FuseFunctionNodes()
223 auto arg_num = std::distance(second_inputs.begin(), input_it); in FuseFunctionNodes()
239 auto input_it = in FuseReturns() local
241 if (input_it == second_inputs.end()) continue; in FuseReturns()
243 auto input_idx = std::distance(second_inputs.begin(), input_it); in FuseReturns()
|
| /external/tensorflow/tensorflow/core/profiler/internal/ |
| D | tfprof_node.h | 711 auto input_it = nodes_map_->find(inp.second); in input_shapes() local
712 if (input_it == nodes_map_->end()) continue; in input_shapes()
716 const TFGraphNode* input_node = input_it->second.get(); in input_shapes()
|