Searched defs:state_tensor (Results 1 – 10 of 10) sorted by relevance
84 Tensor* state_tensor; member
70 Tensor* state_tensor = arg->state_tensor; in operator ()() local
39 Tensor* state_tensor = arg->state_tensor; in operator ()() local
104 auto state_tensor = at::detail::empty_cpu( in get_state() local
158 auto state_tensor = static_cast<SGDParamState&>(*state_kv.second); in ~SGD() local
86 bool state_tensor = false; member
225 …auto state_tensor = at::detail::empty_cpu({(int64_t)size}, ScalarType::Byte, std::nullopt, std::nu… in get_state() local
329 …auto state_tensor = at::detail::empty_cpu({(int64_t)total_size}, ScalarType::Byte, std::nullopt, s… in get_state() local
98 auto state_tensor = gen.get_state(); in THPGenerator_getState() local
992 tensorflow.TensorSpecProto state_tensor = 3; field