Home
last modified time | relevance | path

Searched refs:consumer_op (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/framework/
Dsubscribe.py117 for consumer_op in list(tensor.consumers()): # explicit copy
118 update_input.append((consumer_op, list(consumer_op.inputs).index(tensor)))
132 for consumer_op, index in update_input:
133 consumer_op._update_input(index, out) # pylint: disable=protected-access
135 for consumer_op in update_control_input:
140 new_control_inputs = consumer_op.control_inputs
145 consumer_op._remove_all_control_inputs()
146 consumer_op._add_control_inputs(new_control_inputs)
/external/tensorflow/tensorflow/contrib/graph_editor/
Dtransform.py481 for t, tmp_t_, consumer_op in info.tmp_cyclic_ts:
485 if consumer_op not in info.transformed_ops:
487 consumer_op.name))
489 consumer_op_ = info.transformed_ops[consumer_op]
557 def _transformed_t(self, info, t, consumer_op): argument
572 if consumer_op.type == "Merge":
573 first_input = consumer_op.inputs[0]
574 tmp_t_ = self._transformed_t(info, first_input, consumer_op)
577 tmp_t_ = self._transformed_t(info, enter_input, consumer_op)
584 info.tmp_cyclic_ts.append((t, tmp_t_, consumer_op))
/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantize_weights.cc295 OperatorT* consumer_op = consumer_op_info.op; in QuantizeWeightsInternal() local
297 model->operator_codes[consumer_op->opcode_index]->builtin_code; in QuantizeWeightsInternal()
303 IsHybridEvaluationOp(consumer_op, consumer_op_code) && in QuantizeWeightsInternal()
304 CheckAllOpInputsQuantized(subgraph, consumer_op, consumer_op_code); in QuantizeWeightsInternal()
/external/tensorflow/tensorflow/contrib/graph_editor/tests/
Dmatch.py119 for consumer_op, consumer_op_match in zip(output_t.consumers(),
123 if not consumer_op_match(consumer_op):
/external/tensorflow/tensorflow/python/tpu/
Dtensor_tracer.py578 consumers += [consumer_op for consumer_op in out_tensor.consumers()]
832 for consumer_op in tensor.consumers():
833 if consumer_op.name not in opname_idx_map:
835 'consumer_op %s is not in opname_idx_map'%consumer_op.name)
836 line += ' %d'%opname_idx_map[consumer_op.name]
1513 for consumer_op in consumers:
1515 consumer_op._add_control_input(trace_op)
/external/tensorflow/tensorflow/lite/toco/
Dimport_tensorflow.cc2087 for (const auto& consumer_op : model->operators) { in AddExtraOutputs() local
2088 for (const string& input : consumer_op->inputs) { in AddExtraOutputs()