Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A 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
H A Dreadonly_references_to_resources.cc175 ReadVariableOp read_variable_op = builder.create<ReadVariableOp>( in runOnOperation() local
H A Dresource_op_lifting.cc124 if (auto read_variable_op = dyn_cast<TF::ReadVariableOp>(&op)) { in ForwardStoreToLoad() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dtpu_add_resource_device_attribute.cc103 auto read_variable_op = in runOnOperation() local