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

  /external/tensorflow/tensorflow/core/kernels/
queue_ops.cc 240 const Tensor& Tnum_elements = ctx->input(1);
241 int32 num_elements = Tnum_elements.flat<int32>()(0);
326 const Tensor& Tnum_elements = ctx->input(1);
327 int32 num_elements = Tnum_elements.flat<int32>()(0);
barrier_ops.cc 584 const Tensor* Tnum_elements;
585 OP_REQUIRES_OK_ASYNC(ctx, ctx->input("num_elements", &Tnum_elements),
587 OP_REQUIRES_ASYNC(ctx, TensorShapeUtils::IsScalar(Tnum_elements->shape()),
590 const int32 num_elements = Tnum_elements->scalar<int32>()();

Completed in 121 milliseconds