Searched refs:resource_handle_t (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
D | stats_aggregator_ops.cc | 259 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/ |
D | multi_device_iterator_ops.cc | 663 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()
|
D | iterator_ops.cc | 1048 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()
|