| /external/tensorflow/tensorflow/core/kernels/ |
| D | stateful_random_ops_gpu.cu.cc | 70 Tensor* state_tensor = arg->state_tensor; in operator ()() local
71 OP_REQUIRES(ctx, state_tensor != 0, in operator ()()
79 auto state_tensor_flat = state_tensor->flat<StateElementType>(); in operator ()()
|
| D | stateful_random_ops.cc | 39 Tensor* state_tensor = arg->state_tensor; in operator ()() local
41 auto state_tensor_flat = state_tensor->flat<StateElementType>(); in operator ()()
117 arg.state_tensor = var_tensor; in UpdateVariableAndFill()
|
| D | stateful_random_ops_cpu_gpu.h | 84 Tensor* state_tensor; member
|
| /external/pytorch/aten/src/ATen/xpu/ |
| D | XPUGeneratorImpl.cpp | 104 auto state_tensor = at::detail::empty_cpu( in get_state() local
111 auto rng_state = state_tensor.data_ptr<uint8_t>(); in get_state()
117 return state_tensor.getIntrusivePtr(); in get_state()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | operator_property.cc | 319 tensor_property_19.state_tensor = true; in GetOperatorProperty()
388 tensor_property_19.state_tensor = true; in GetOperatorProperty()
452 tensor_property_19.state_tensor = true; in GetOperatorProperty()
520 tensor_property_19.state_tensor = true; in GetOperatorProperty()
595 tensor_property_19.state_tensor = true; in GetOperatorProperty()
662 tensor_property_19.state_tensor = true; in GetOperatorProperty()
725 tensor_property_19.state_tensor = true; in GetOperatorProperty()
791 tensor_property_19.state_tensor = true; in GetOperatorProperty()
1048 tensor_property_state.state_tensor = true; in GetOperatorProperty()
|
| D | operator_property.h | 86 bool state_tensor = false; member
|
| D | quantize_model.cc | 990 tensor_property.state_tensor) { in QuantizeOpInput()
|
| /external/executorch/extension/training/optimizer/ |
| D | sgd.cpp | 158 auto state_tensor = static_cast<SGDParamState&>(*state_kv.second); in ~SGD() local
159 free(state_tensor.momentum_buffer().unsafeGetTensorImpl()->mutable_data()); in ~SGD()
161 delete state_tensor.momentum_buffer().unsafeGetTensorImpl(); in ~SGD()
|
| /external/pytorch/aten/src/ATen/mps/ |
| D | MPSGeneratorImpl.mm | 71 auto state_tensor = at::detail::empty_cpu(
73 auto rng_state = state_tensor.data_ptr<uint8_t>();
84 return state_tensor.getIntrusivePtr();
|
| /external/pytorch/aten/src/ATen/ |
| D | CPUGeneratorImpl.cpp | 225 …auto state_tensor = at::detail::empty_cpu({(int64_t)size}, ScalarType::Byte, std::nullopt, std::nu… in get_state() local
226 auto rng_state = state_tensor.data_ptr(); in get_state()
252 return state_tensor.getIntrusivePtr(); in get_state()
|
| /external/pytorch/aten/src/ATen/cuda/ |
| D | CUDAGeneratorImpl.cpp | 329 …auto state_tensor = at::detail::empty_cpu({(int64_t)total_size}, ScalarType::Byte, std::nullopt, s… in get_state() local
330 auto rng_state = state_tensor.data_ptr<uint8_t>(); in get_state()
336 return state_tensor.getIntrusivePtr(); in get_state()
|
| /external/pytorch/torch/csrc/ |
| D | Generator.cpp | 98 auto state_tensor = gen.get_state(); in THPGenerator_getState() local
100 return THPVariable_Wrap(std::move(state_tensor)); in THPGenerator_getState()
|
| /external/pytorch/test/inductor/ |
| D | test_compiled_optimizers.py | 467 state_tensor = opt_compiled.state[
472 ] = torch.zeros_like(state_tensor)
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_quantize_helper.h | 338 if (tensor_property.state_tensor && !stats_op.getResult().hasOneUse()) { in replaceStatsOp()
|
| /external/federated-compute/fcp/artifact_building/ |
| D | federated_compute_plan_builder.py | 552 state_tensor=tensor_spec.experimental_as_proto()
|
| /external/federated-compute/fcp/protos/ |
| D | plan.proto | 992 tensorflow.TensorSpecProto state_tensor = 3; field
|