HomeSort by relevance Sort by last modified time
    Searched refs:dst_dtype_ (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
cast_op.cc 57 OP_REQUIRES_OK(ctx, ctx->GetAttr("DstT", &dst_dtype_));
73 DataTypeString(dst_dtype_), " is not supported");
81 if (src_dtype_ == dst_dtype_) {
86 work_ = GetCpuCastFromBool(dst_dtype_);
88 work_ = GetCpuCastFromUint8(dst_dtype_);
90 work_ = GetCpuCastFromInt8(dst_dtype_);
92 work_ = GetCpuCastFromUint16(dst_dtype_);
94 work_ = GetCpuCastFromInt16(dst_dtype_);
96 work_ = GetCpuCastFromInt32(dst_dtype_);
98 work_ = GetCpuCastFromInt64(dst_dtype_);
    [all...]
cast_op.h 38 DataType dst_dtype_; member in class:tensorflow::CastOpBase
  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
cast_op.cc 30 OP_REQUIRES_OK(ctx, ctx->GetAttr("DstT", &dst_dtype_));
32 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(dst_dtype_, &dst_type_));
40 if (src_dtype_ == dst_dtype_) {
42 } else if (dst_dtype_ == DT_BOOL) {
57 DataType src_dtype_, dst_dtype_; member in class:tensorflow::__anon38730::CastOp

Completed in 163 milliseconds