Home
last modified time | relevance | path

Searched defs:read_variable_op (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dreadonly_references_to_resources.cc175 ReadVariableOp read_variable_op = builder.create<ReadVariableOp>( in runOnOperation() local
Dtf_saved_model_freeze_variables.cc85 void PropagateUsage(TF::ReadVariableOp read_variable_op, ElementsAttr value) { in PropagateUsage()
104 if (auto read_variable_op = dyn_cast<TF::ReadVariableOp>(user_op)) { in PropagateUsage() local
Dresource_op_lifting.cc124 if (auto read_variable_op = dyn_cast<TF::ReadVariableOp>(&op)) { in ForwardStoreToLoad() local
/external/tensorflow/tensorflow/dtensor/mlir/
Dtpu_add_resource_device_attribute.cc103 auto read_variable_op = in runOnOperation() local