Searched refs:OptionalVariant (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | optional_ops.cc | 30 const OptionalVariant& from, OptionalVariant* to, in OptionalDeviceCopy()
49 *to = OptionalVariant(std::move(to_values)); in OptionalDeviceCopy()
58 OptionalVariant, DIRECTION, OptionalDeviceCopy)
64 REGISTER_UNARY_VARIANT_DECODE_FUNCTION(OptionalVariant,
89 const OptionalVariant* optional = in Compute()
90 optional_input->scalar<Variant>()().get<OptionalVariant>(); in Compute()
107 const OptionalVariant* optional = in Compute()
108 optional_input->scalar<Variant>()().get<OptionalVariant>(); in Compute()
140 OptionalVariant v(std::move(value)); in WriteOptionalWithValueToOutput()
151 OptionalVariant v; in WriteOptionalNoneToOutput()
[all …]
|
| D | optional_ops.h | 40 class OptionalVariant {
43 OptionalVariant() : values_(nullptr) {} in OptionalVariant() function
47 explicit OptionalVariant(std::vector<Tensor> values) { in OptionalVariant() function
51 OptionalVariant(const OptionalVariant& other) : values_(other.values_) {} in OptionalVariant() function
108 Status OptionalZerosLike(OpKernelContext* ctx, const OptionalVariant& x, in OptionalZerosLike()
109 OptionalVariant* y) { in OptionalZerosLike()
120 *y = OptionalVariant(zero_tensors); in OptionalZerosLike()
125 Status OptionalBinaryAdd(OpKernelContext* ctx, const OptionalVariant& a, in OptionalBinaryAdd()
126 const OptionalVariant& b, OptionalVariant* out) { in OptionalBinaryAdd()
151 *out = OptionalVariant(out_tensors); in OptionalBinaryAdd()
|
| D | optional_ops.cu.cc | 27 DEVICE_GPU, OptionalVariant,
31 OptionalVariant,
|