HomeSort by relevance Sort by last modified time
    Searched refs:IsRefType (Results 1 - 25 of 37) sorted by null

1 2

  /external/tensorflow/tensorflow/core/kernels/
identity_op.h 28 if (IsRefType(context->input_dtype(0))) {
assign_op.h 37 OP_REQUIRES(context, IsRefType(context->input_type(0)),
logging_ops.cc 64 if (IsRefType(ctx->input_dtype(0))) {
ops_testutil.h 191 CHECK(!IsRefType(context_->input_dtype(input_index)));
212 bool is_ref = IsRefType(input_types_[inputs_.size()]);
variable_ops.cc 135 OP_REQUIRES(context, IsRefType(context->input_type(0)),
146 CHECK(IsRefType(context->input_dtype(0)));
control_flow_ops.cc 226 if (IsRefType(context->input_dtype(i))) {
337 if (IsRefType(context->input_dtype(0))) {
432 if (IsRefType(context->input_dtype(0))) {
511 if (IsRefType(context->input_dtype(0))) {
scatter_nd_op.cc 92 } else if (IsRefType(c->input_type(0))) {
113 DCHECK(IsRefType(c->input_dtype(0)));
145 } else if (IsRefType(c->input_dtype(0))) {
resize_area_op_test.cc 41 bool is_ref = IsRefType(input_types_[inputs_.size()]);
resize_bicubic_op_test.cc 41 bool is_ref = IsRefType(input_types_[inputs_.size()]);
  /external/tensorflow/tensorflow/core/framework/
types.cc 109 if (IsRefType(dtype)) {
120 if (DataTypeFromString(sp, &non_ref) && !IsRefType(non_ref)) {
types.h 238 inline bool IsRefType(DataType dtype) {
242 DCHECK(!IsRefType(dtype));
246 DCHECK(IsRefType(dtype));
250 return IsRefType(dtype) ? RemoveRefType(dtype) : dtype;
op_compatibility_test.cc 67 if (IsRefType(old_in_types[i]) && !IsRefType(new_in_types[i])) {
76 if (!IsRefType(old_out_types[i]) && IsRefType(new_out_types[i])) {
    [all...]
attr_value_util.cc 211 if (IsRefType(attr_value.type())) {
226 if (IsRefType(dtype)) {
node_def_builder.cc 188 if (input_arg->is_ref() && !IsRefType(dt)) {
op_kernel.cc 655 DCHECK(!IsRefType(type));
723 DCHECK(!IsRefType(params_->op_kernel->output_type(index)));
749 DCHECK(IsRefType(params_->op_kernel->output_type(index)));
    [all...]
  /external/tensorflow/tensorflow/contrib/framework/kernels/
zero_initializer_op.cc 36 OP_REQUIRES(ctx, IsRefType(ctx->input_type(0)),
  /external/tensorflow/tensorflow/compiler/jit/kernels/
parallel_check_op.cc 129 if (IsRefType(ctx->input_dtype(i))) {
  /external/tensorflow/tensorflow/core/graph/
subgraph_test.cc 210 EXPECT_FALSE(IsRefType(CHECK_NOTNULL(n)->output_type(0)));
222 EXPECT_FALSE(IsRefType(CHECK_NOTNULL(n)->output_type(0)));
optimizer_cse.cc 126 if (IsRefType(dt)) return true;
graph_partition.cc 349 if ((IsEnter(node) && !IsRefType(node->input_type(0))) ||
    [all...]
  /external/tensorflow/tensorflow/core/common_runtime/
memory_types.cc 180 const bool has_ref = IsRefType(e->src()->output_type(e->src_output()));
placer.cc 635 !IsRefType(node->output_type(0));
681 if (input_type == DT_RESOURCE || IsRefType(input_type)) {
    [all...]
  /external/tensorflow/tensorflow/core/grappler/optimizers/
graph_rewriter.cc 162 IsRefType(fanin_info->outputs[position])) {
memory_optimizer.cc 705 if (IsRefType(input_type)) {
    [all...]
  /external/tensorflow/tensorflow/core/util/
mkl_util.h     [all...]

Completed in 566 milliseconds

1 2