Searched refs:tensor_lengths (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/python/tpu/ |
D | feature_column_v2.py | 576 tensor_lengths = inputs.get(self.get_sequence_length_feature_key_name()) 580 tensor_lengths = array_ops.squeeze(tensor_lengths, -1) 589 dense_tensor=tensor, sequence_length=tensor_lengths) 604 tensor_lengths = transformation_cache.get( 611 tensor_lengths = array_ops.squeeze(tensor_lengths, -1) 614 dense_tensor=tensor, sequence_length=tensor_lengths) 742 tensor_lengths = transformation_cache.get( 749 tensor_lengths = array_ops.squeeze(tensor_lengths, -1) 752 dense_tensor=tensor, sequence_length=tensor_lengths)
|
D | feature_column.py | 489 tensor_lengths = inputs.get(self.get_sequence_length_feature_key_name()) 493 tensor_lengths = array_ops.squeeze(tensor_lengths, -1) 502 dense_tensor=tensor, sequence_length=tensor_lengths) 627 tensor_lengths = inputs.get(self.get_sequence_length_feature_key_name()) 636 dense_tensor=tensor, sequence_length=tensor_lengths)
|
/external/tensorflow/tensorflow/core/kernels/ |
D | tensor_array_ops.cc | 1236 const Tensor* tensor_lengths; in Compute() local 1237 OP_REQUIRES_OK(ctx, ctx->input("lengths", &tensor_lengths)); in Compute() 1239 OP_REQUIRES(ctx, TensorShapeUtils::IsVector(tensor_lengths->shape()), in Compute() 1242 tensor_lengths->shape().DebugString())); in Compute() 1244 FastBoundsCheck(tensor_lengths->NumElements(), in Compute() 1249 int32 num_tensors = static_cast<int32>(tensor_lengths->NumElements()); in Compute() 1250 auto tensor_lengths_t = tensor_lengths->vec<int64>(); in Compute()
|
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | tensor_list_ops.cc | 493 int64 tensor_lengths = element_dims[1]; in Compile() local 495 std::vector<int64> new_dims = {num_elements * tensor_lengths}; in Compile() 505 xla::XlaOp lengths = xla::ConstantR1(b, num_elements, tensor_lengths); in Compile()
|