Home
last modified time | relevance | path

Searched refs:state_tensor (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dstateful_random_ops_gpu.cu.cc70 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 ()()
Dstateful_random_ops.cc39 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()
Dstateful_random_ops_cpu_gpu.h83 Tensor* state_tensor; member
/external/tensorflow/tensorflow/lite/tools/optimize/
Doperator_property.cc298 tensor_property_19.state_tensor = true; in GetOperatorProperty()
367 tensor_property_19.state_tensor = true; in GetOperatorProperty()
431 tensor_property_19.state_tensor = true; in GetOperatorProperty()
499 tensor_property_19.state_tensor = true; in GetOperatorProperty()
574 tensor_property_19.state_tensor = true; in GetOperatorProperty()
641 tensor_property_19.state_tensor = true; in GetOperatorProperty()
704 tensor_property_19.state_tensor = true; in GetOperatorProperty()
770 tensor_property_19.state_tensor = true; in GetOperatorProperty()
992 tensor_property_state.state_tensor = true; in GetOperatorProperty()
Doperator_property.h83 bool state_tensor = false; member
Dquantize_model.cc711 tensor_property.state_tensor) { in QuantizeOpInput()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_quantize_helper.h338 if (tensor_property.state_tensor && !stats_op.getResult().hasOneUse()) { in replaceStatsOp()