Home
last modified time | relevance | path

Searched refs:tensor_lengths (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/tpu/
Dfeature_column_v2.py576 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)
Dfeature_column.py489 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/
Dtensor_array_ops.cc1236 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/
Dtensor_list_ops.cc493 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()