Home
last modified time | relevance | path

Searched defs:state_tensor (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dstateful_random_ops_cpu_gpu.h84 Tensor* state_tensor; member
H A Dstateful_random_ops_gpu.cu.cc70 Tensor* state_tensor = arg->state_tensor; in operator ()() local
H A Dstateful_random_ops.cc39 Tensor* state_tensor = arg->state_tensor; in operator ()() local
/aosp_15_r20/external/pytorch/aten/src/ATen/xpu/
H A DXPUGeneratorImpl.cpp104 auto state_tensor = at::detail::empty_cpu( in get_state() local
/aosp_15_r20/external/executorch/extension/training/optimizer/
H A Dsgd.cpp158 auto state_tensor = static_cast<SGDParamState&>(*state_kv.second); in ~SGD() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/
H A Doperator_property.h86 bool state_tensor = false; member
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DCPUGeneratorImpl.cpp225 …auto state_tensor = at::detail::empty_cpu({(int64_t)size}, ScalarType::Byte, std::nullopt, std::nu… in get_state() local
/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/
H A DCUDAGeneratorImpl.cpp329 …auto state_tensor = at::detail::empty_cpu({(int64_t)total_size}, ScalarType::Byte, std::nullopt, s… in get_state() local
/aosp_15_r20/external/pytorch/torch/csrc/
H A DGenerator.cpp98 auto state_tensor = gen.get_state(); in THPGenerator_getState() local
/aosp_15_r20/external/federated-compute/fcp/protos/
H A Dplan.proto992 tensorflow.TensorSpecProto state_tensor = 3; field
/aosp_15_r20/out/soong/.intermediates/external/federated-compute/federated-compute-cc-proto-lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/fcp/protos/
Dplan.pb.cc13779 ServerAggregationConfig_IntrinsicArg::_Internal::state_tensor(const ServerAggregationConfig_Intrins… in state_tensor() function in google::internal::federated::plan::ServerAggregationConfig_IntrinsicArg::_Internal
13806 …egationConfig_IntrinsicArg::set_allocated_state_tensor(::tensorflow::TensorSpecProto* state_tensor… in set_allocated_state_tensor()
Dplan.pb.h16755 inline const ::tensorflow::TensorSpecProto& ServerAggregationConfig_IntrinsicArg::state_tensor() co… in state_tensor() function
16770 …_IntrinsicArg::unsafe_arena_set_allocated_state_tensor(::tensorflow::TensorSpecProto* state_tensor… in unsafe_arena_set_allocated_state_tensor()