Searched defs:padded_shape (Results 1 – 7 of 7) sorted by relevance
66 Shape padded_shape = input_array.shape(); in Slice() local
69 xla::Shape padded_shape; in TFE_TensorHandleTensorDebugInfo() local
206 def padded_batch_window(dataset, padded_shape, padding_value=None): argument241 def _padded_batch_dense_window(dataset, padded_shape, padding_value=None): argument306 def _padded_batch_sparse_window(dataset, padded_shape): argument
64 PartialTensorShape padded_shape; in MakeDataset() local275 const PartialTensorShape& padded_shape = in GetNextInternal() local
54 std::vector<int64> padded_shape(input_shape.begin(), input_shape.end()); in SpaceToBatch() local
712 TensorShape padded_shape; in Compute() local
2790 def _is_padded_shape_compatible_with(padded_shape, input_component_shape): argument2814 def _padded_shape_to_tensor(padded_shape, input_component_shape): argument