Home
last modified time | relevance | path

Searched defs:padded_shape (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_constant_slice.cc66 Shape padded_shape = input_array.shape(); in Slice() local
/external/tensorflow/tensorflow/c/eager/
Dc_api_debug.cc69 xla::Shape padded_shape; in TFE_TensorHandleTensorDebugInfo() local
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dbatching.py206 def padded_batch_window(dataset, padded_shape, padding_value=None): argument
241 def _padded_batch_dense_window(dataset, padded_shape, padding_value=None): argument
306 def _padded_batch_sparse_window(dataset, padded_shape): argument
/external/tensorflow/tensorflow/core/kernels/data/
Dpadded_batch_dataset_op.cc64 PartialTensorShape padded_shape; in MakeDataset() local
275 const PartialTensorShape& padded_shape = in GetNextInternal() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dspacetobatch_op.cc54 std::vector<int64> padded_shape(input_shape.begin(), input_shape.end()); in SpaceToBatch() local
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_fused_image_transform.cc712 TensorShape padded_shape; in Compute() local
/external/tensorflow/tensorflow/python/data/ops/
Ddataset_ops.py2790 def _is_padded_shape_compatible_with(padded_shape, input_component_shape): argument
2814 def _padded_shape_to_tensor(padded_shape, input_component_shape): argument