Home
last modified time | relevance | path

Searched defs:normalized_shape (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_normalization.cc69 auto normalized_shape = Normalize(shape); in DefaultAction() local
121 auto normalized_shape = in HandleConcatenate() local
152 auto normalized_shape = Normalize(s); in HandleBroadcast() local
300 auto normalized_shape = Normalize(s); in HandleTranspose() local
/external/XNNPACK/src/
Dnormalization.c76 size_t* normalized_shape, in xnn_normalize_transpose_permutation()
/external/XNNPACK/src/operators/
Dtranspose-nd.c189 size_t normalized_shape[XNN_MAX_TENSOR_DIMS]; in setup_transpose_nd() local
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Doptimize.cc851 int64_t normalized_shape[2] = {element_size, 1}; in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.cc1713 Shape normalized_shape = MergeDimensions(segments, descending_layout_shape); in FindTranspose021() local