HomeSort by relevance Sort by last modified time
    Searched defs:src_dtype_ (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
cast_op.cc 29 OP_REQUIRES_OK(ctx, ctx->GetAttr("SrcT", &src_dtype_));
31 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(src_dtype_, &src_type_));
40 if (src_dtype_ == dst_dtype_) {
43 output = builder->Ne(input, XlaHelpers::Zero(builder, src_dtype_));
57 DataType src_dtype_, dst_dtype_; member in class:tensorflow::__anon38730::CastOp
  /external/tensorflow/tensorflow/core/kernels/
cast_op.h 37 DataType src_dtype_; member in class:tensorflow::CastOpBase

Completed in 80 milliseconds