Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_slice_op.cc222 std::shared_ptr<primitive> reorder_prim; member
459 MklSlicePrimitive<T>* reorder_prim = in ComputeMklSlice() local
H A Dmkl_quantize_op.cc117 std::shared_ptr<primitive> reorder_prim; member
482 MklReorderWithScalePrimitive* reorder_prim = in Compute() local
H A Dmkl_requantize_per_channel_op.cc146 std::unique_ptr<dnnl::primitive> reorder_prim( in Compute() local