Home
last modified time | relevance | path

Searched refs:resource_handle_t (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dstats_aggregator_ops.cc259 const Tensor& resource_handle_t = ctx->input(0); in Compute() local
260 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(resource_handle_t.shape()), in Compute()
281 const Tensor& resource_handle_t = ctx->input(0); in Compute() local
282 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(resource_handle_t.shape()), in Compute()
/external/tensorflow/tensorflow/core/kernels/data/
Dmulti_device_iterator_ops.cc663 const Tensor& resource_handle_t = ctx->input(0); in Compute() local
664 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(resource_handle_t.shape()), in Compute()
677 resource_handle_t.scalar<ResourceHandle>()().SerializeAsString(); in Compute()
731 Tensor* resource_handle_t; in Compute() local
733 ctx, ctx->allocate_output(0, TensorShape({}), &resource_handle_t)); in Compute()
734 resource_handle_t->scalar<ResourceHandle>()() = resource_handle; in Compute()
Diterator_ops.cc1048 const Tensor& resource_handle_t = ctx->input(0); in Compute() local
1049 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(resource_handle_t.shape()), in Compute()
1063 resource_handle_t.scalar<ResourceHandle>()().SerializeAsString(); in Compute()
1112 Tensor* resource_handle_t; in Compute() local
1114 ctx->allocate_output(0, TensorShape({}), &resource_handle_t)); in Compute()
1115 resource_handle_t->scalar<ResourceHandle>()() = resource_handle; in Compute()
1132 const Tensor& resource_handle_t = ctx->input(0); in Compute() local
1133 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(resource_handle_t.shape()), in Compute()