Searched refs:key_tensor (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | map_stage_op.cc | 521 const Tensor* key_tensor; in Compute() local
525 OP_REQUIRES_OK(ctx, ctx->input("key", &key_tensor)); in Compute()
530 Tensor key(*key_tensor); in Compute()
583 const Tensor* key_tensor; in Compute() local
586 OP_REQUIRES_OK(ctx, ctx->input("key", &key_tensor)); in Compute()
588 OP_REQUIRES_OK(ctx, map->pop(key_tensor, indices_tensor, &tuple)); in Compute()
644 const Tensor* key_tensor; in Compute() local
647 OP_REQUIRES_OK(ctx, ctx->input("key", &key_tensor)); in Compute()
649 OP_REQUIRES_OK(ctx, map->get(key_tensor, indices_tensor, &tuple)); in Compute()
|
| /external/tensorflow/tensorflow/contrib/hadoop/kernels/ |
| D | hadoop_dataset_ops.cc | 264 Tensor key_tensor(ctx->allocator({}), DT_STRING, {}); in GetNextInternal() local
265 key_tensor.scalar<string>()() = key; in GetNextInternal()
266 out_tensors->emplace_back(std::move(key_tensor)); in GetNextInternal()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | lmdb_dataset_op.cc | 96 Tensor& key_tensor = out_tensors->back(); in GetNextInternal() local
97 key_tensor.scalar<string>()() = string( in GetNextInternal()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | lookup_ops.py | 211 key_tensor = keys
213 key_tensor = keys.values
221 (self.resource_handle, key_tensor, self._default_value)):
223 self.resource_handle, key_tensor, self._default_value)
225 values.set_shape(key_tensor.get_shape())
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 514 Tensor key_tensor = ctx->input(ctx->num_inputs() - 1); in Compute() local
515 const XlaExecutableClosureStore::KeyT& key = key_tensor.flat<string>()(0); in Compute()
|