Home
last modified time | relevance | path

Searched defs:padding_shape (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.h187 const Shape padding_shape = variable
/external/tensorflow/tensorflow/cc/gradients/
Darray_grad.cc490 auto padding_shape = Stack(scope, {input_rank, 1}); in SliceGrad() local
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_builder.cc2843 auto padding_shape = padding.shape; in Parse() local