Searched refs:output_ops (Results 1 – 2 of 2) sorted by relevance
374 def _get_dependent_variables(input_ops, output_ops): argument386 output_ops = nest.map_structure(gen_array_ops.identity, output_ops)388 seed_ops=output_ops,465 input_ops=filtered_input_tensors, output_ops=flat_result)
1179 llvm::SmallVector<mlir::TF::MulOp, 4> output_ops; in ExpandOp() local1213 output_ops.emplace_back(global_shape); in ExpandOp()1228 return output_ops.empty() ? op : output_ops[0].getOperation(); in ExpandOp()