Home
last modified time | relevance | path

Searched refs:WriteLocalVariable (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dstack_ops_decomposition.cc406 cutil::WriteLocalVariable(local_size_var, in HandleStackV2Op()
409 cutil::WriteLocalVariable(local_var, buffer, builder, stack.getLoc()); in HandleStackV2Op()
434 cutil::WriteLocalVariable(push.handle(), stack_val, builder, push.getLoc()); in HandleStackPushV2Op()
438 cutil::WriteLocalVariable(it->getSecond(), index, builder, push.getLoc()); in HandleStackPushV2Op()
461 cutil::WriteLocalVariable(it->getSecond(), new_size, builder, pop.getLoc()); in HandleStackPopV2Op()
Dcollection_ops_util.h89 TF::AssignVariableOp WriteLocalVariable(Value local_var, Value value,
Dtensor_array_ops_decomposition.cc213 cutil::WriteLocalVariable(local_var, buffer, builder, ta.getLoc()); in HandleTensorArrayV3Op()
279 cutil::WriteLocalVariable(local_var, buffer, builder, write.getLoc()); in HandleTensorArrayWriteV3Op()
354 cutil::WriteLocalVariable(local_var, buffer, builder, split.getLoc()); in HandleTensorArraySplitV3Op()
394 cutil::WriteLocalVariable(*var, buffer, builder, op->getLoc()); in CreateAndInitializeGradVariable()
454 cutil::WriteLocalVariable(local_var, buffer, builder, scatter.getLoc()); in HandleTensorArrayScatterV3Op()
Dcollection_ops_util.cc226 TF::AssignVariableOp WriteLocalVariable(Value local_var, Value value, in WriteLocalVariable() function