Home
last modified time | relevance | path

Searched refs:new_outputs (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/python/framework/
Dop_callbacks.py200 new_outputs = outputs
202 new_outputs = callback(
206 new_outputs,
209 if new_outputs is not None and len(new_outputs) != len(outputs):
213 (len(new_outputs), op_name, len(outputs)))
214 return new_outputs
/external/mesa3d/src/compiler/nir/
Dnir_lower_io_to_temporaries.c42 struct exec_list new_outputs; member
97 emit_copies(&b, &state->new_outputs, &state->old_outputs); in emit_output_copies_impl()
103 emit_copies(&b, &state->old_outputs, &state->new_outputs); in emit_output_copies_impl()
111 emit_copies(&b, &state->new_outputs, &state->old_outputs); in emit_output_copies_impl()
347 exec_list_make_empty(&state.new_outputs); in nir_lower_io_to_temporaries()
354 exec_list_push_tail(&state.new_outputs, &output->node); in nir_lower_io_to_temporaries()
381 exec_list_append(&shader->variables, &state.new_outputs); in nir_lower_io_to_temporaries()
Dnir_lower_io_to_vector.c384 nir_variable *new_outputs[MAX_SLOTS][4] = {{0}}; in nir_lower_io_to_vector_impl() local
405 new_outputs, flat_outputs)) in nir_lower_io_to_vector_impl()
447 new_outputs[loc][old_frac]; in nir_lower_io_to_vector_impl()
500 nir_variable *new_var = new_outputs[loc][old_frac]; in nir_lower_io_to_vector_impl()
/external/tensorflow/tensorflow/python/saved_model/model_utils/
Dexport_output.py295 new_outputs = {}
298 new_outputs[key] = val
299 return new_outputs
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dpfor.py632 new_outputs = []
648 new_outputs.append(new_output)
649 return new_outputs
744 new_outputs = self._process_body(pfor_input, inputs_stacked,
752 new_args = ([not_all_done, new_indices] + new_outputs +
1555 new_outputs = reduce_output
1563 new_outputs = y_op
1565 new_outputs = [wrap(x, False) for x in y_op.outputs]
1575 new_outputs = new_op
1577 new_outputs = []
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dxla_sharding_util.cc433 llvm::SmallVector<mlir::Value, 4> new_outputs; in HandleTileShardedOutputs() local
434 new_outputs.reserve(num_splits); in HandleTileShardedOutputs()
446 new_outputs.emplace_back(concat_op.getResult()); in HandleTileShardedOutputs()
449 std::swap(new_outputs, outputs_to_merge); in HandleTileShardedOutputs()
/external/mesa3d/src/freedreno/ir3/
Dir3_nir_lower_tess.c43 struct exec_list new_outputs; member
930 exec_list_make_empty(&state.new_outputs); in ir3_nir_lower_gs()
937 exec_list_push_tail(&state.new_outputs, &output->node); in ir3_nir_lower_gs()
983 foreach_two_lists(dest_node, &state.new_outputs, src_node, &state.emit_outputs) { in ir3_nir_lower_gs()
992 exec_list_append(&shader->variables, &state.new_outputs); in ir3_nir_lower_gs()
/external/tensorflow/tensorflow/python/ops/
Dwhile_v2.py370 new_outputs = body_graph.outputs[orig_num_params:]
382 while_op._add_outputs([t.dtype for t in new_outputs],
383 [t.shape for t in new_outputs])
384 _copy_handle_data(new_outputs, while_op.outputs[orig_num_params:])
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtensor_list_ops_decomposition.cc135 auto new_outputs = llvm::to_vector<8>(old_terminator->getOperands()); in AddTensorListSizesToTerminator() local
141 output_buffer_to_size.emplace_back(retval.index(), new_outputs.size(), in AddTensorListSizesToTerminator()
143 new_outputs.push_back(it->getSecond().size); in AddTensorListSizesToTerminator()
146 .create<TerminatorOp>(old_terminator->getLoc(), new_outputs); in AddTensorListSizesToTerminator()