Searched refs:resource_handle_t (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | multi_device_iterator_ops.cc | 580 const Tensor& resource_handle_t = ctx->input(0); in Compute() local
581 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(resource_handle_t.shape()), in Compute()
595 resource_handle_t.scalar<ResourceHandle>()().SerializeAsString(); in Compute()
650 Tensor* resource_handle_t; in Compute() local
652 ctx, ctx->allocate_output(0, TensorShape({}), &resource_handle_t)); in Compute()
653 resource_handle_t->scalar<ResourceHandle>()() = resource_handle; in Compute()
|
| D | iterator_ops.cc | 1051 const Tensor& resource_handle_t = ctx->input(0); in Compute() local
1052 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(resource_handle_t.shape()), in Compute()
1066 resource_handle_t.scalar<ResourceHandle>()().SerializeAsString(); in Compute()
1115 Tensor* resource_handle_t; in Compute() local
1117 ctx->allocate_output(0, TensorShape({}), &resource_handle_t)); in Compute()
1118 resource_handle_t->scalar<ResourceHandle>()() = resource_handle; in Compute()
1128 const Tensor& resource_handle_t = ctx->input(0); in Compute() local
1129 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(resource_handle_t.shape()), in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | stats_aggregator_ops.cc | 127 const Tensor& resource_handle_t = ctx->input(0); in Compute() local
128 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(resource_handle_t.shape()), in Compute()
|