Searched refs:Tnum_elements (Results 1 – 2 of 2) sorted by relevance
211 const Tensor& Tnum_elements = ctx->input(1); in ComputeAsync() local212 int32 num_elements = Tnum_elements.flat<int32>()(0); in ComputeAsync()284 const Tensor& Tnum_elements = ctx->input(1); in ComputeAsync() local285 int32 num_elements = Tnum_elements.flat<int32>()(0); in ComputeAsync()
583 const Tensor* Tnum_elements; in ComputeAsync() local584 OP_REQUIRES_OK_ASYNC(ctx, ctx->input("num_elements", &Tnum_elements), in ComputeAsync()586 OP_REQUIRES_ASYNC(ctx, TensorShapeUtils::IsScalar(Tnum_elements->shape()), in ComputeAsync()589 const int32 num_elements = Tnum_elements->scalar<int32>()(); in ComputeAsync()