Searched refs:state_vars (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_variable_runtime_reformatting.cc | 338 llvm::SmallVector<TF::VarHandleOp, 4> state_vars; in CreateStateVars() local
347 state_vars.push_back(builder->create<TF::VarHandleOp>( in CreateStateVars()
360 return state_vars; in CreateStateVars()
445 auto state_vars = in HandleReplicateOp() local
447 replicate = AddInputsToReplicateOp(replicate, state_vars, devices); in HandleReplicateOp()
475 llvm::SmallVector<Value, 4> state_var_vals(state_vars.size()); in HandleReplicateOp()
476 for (const auto& entry : llvm::enumerate(state_vars)) { in HandleReplicateOp()
|
| /external/tensorflow/tensorflow/python/autograph/operators/ |
| D | control_flow.py | 588 state_vars = get_state()
589 state_vars = tuple(
591 for name, var in zip(symbol_names, state_vars))
592 return (iterate,) + state_vars
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | serialization_base.fbs | 47 state_vars:[StateVariable];
|
| D | serialization_base_generated.h | 668 …const flatbuffers::Vector<flatbuffers::Offset<tflite::gpu::data::StateVariable>> *state_vars() con… in state_vars() function
677 verifier.VerifyVector(state_vars()) && in Verify()
678 verifier.VerifyVectorOfTables(state_vars()) && in Verify()
688 …s::Offset<flatbuffers::Vector<flatbuffers::Offset<tflite::gpu::data::StateVariable>>> state_vars) { in add_state_vars()
689 fbb_.AddOffset(GPUObjectDescriptor::VT_STATE_VARS, state_vars); in add_state_vars()
707 …:Offset<flatbuffers::Vector<flatbuffers::Offset<tflite::gpu::data::StateVariable>>> state_vars = 0,
710 builder_.add_state_vars(state_vars);
717 const std::vector<flatbuffers::Offset<tflite::gpu::data::StateVariable>> *state_vars = nullptr,
719 …auto state_vars__ = state_vars ? _fbb.CreateVector<flatbuffers::Offset<tflite::gpu::data::StateVar…
|
| /external/tensorflow/tensorflow/python/autograph/converters/ |
| D | control_flow.py | 107 state_vars=tuple(block_vars))
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | serialization.cc | 352 for (auto state_fb : *fb_obj->state_vars()) { in Decode()
|