Searched refs:extra_operands (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | while_loop_outline.cc | 138 llvm::SetVector<Value> extra_operands; in OutlineWhile() local
142 extra_operands.insert(new_extern.begin(), new_extern.end()); in OutlineWhile()
146 if (extra_operands.empty() && IsAlreadyOutlined(while_op)) return; in OutlineWhile()
150 types.reserve(extra_operands.size() + while_op.getNumOperands()); in OutlineWhile()
242 if (extra_operands.empty()) return; in OutlineWhile()
244 const int operands_size = while_op.getNumOperands() + extra_operands.size(); in OutlineWhile()
248 operands.append(extra_operands.begin(), extra_operands.end()); in OutlineWhile()
253 for (auto extra_operand : extra_operands) in OutlineWhile()
|
| /external/mesa3d/src/gallium/drivers/zink/nir_to_spirv/ |
| D | spirv_builder.c | 183 SpvDecoration decoration, const uint32_t extra_operands[], in emit_decoration() argument
192 spirv_buffer_emit_word(&b->decorations, extra_operands[i]); in emit_decoration()
298 SpvDecoration decoration, const uint32_t extra_operands[], in emit_member_decoration() argument
309 spirv_buffer_emit_word(&b->decorations, extra_operands[i]); in emit_member_decoration()
660 SpvId extra_operands[5]; in spirv_builder_emit_image_sample() local
663 extra_operands[++num_extra_operands] = bias; in spirv_builder_emit_image_sample()
667 extra_operands[++num_extra_operands] = lod; in spirv_builder_emit_image_sample()
670 extra_operands[++num_extra_operands] = dx; in spirv_builder_emit_image_sample()
671 extra_operands[++num_extra_operands] = dy; in spirv_builder_emit_image_sample()
675 extra_operands[++num_extra_operands] = offset; in spirv_builder_emit_image_sample()
[all …]
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/ |
| D | np_math_ops_test.py | 55 extra_operands=None, argument
85 if extra_operands is not None:
86 for operand1, operand2 in extra_operands:
90 extra_operands = [([1, 2], [[5, 6, 7], [8, 9, 10]]),
94 np_math_ops.dot, np.dot, 'dot', extra_operands=extra_operands)
|