Searched defs:GetPrimitive (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ | ||
| D | mkl_quantize_op.cc | 88 std::shared_ptr<primitive> GetPrimitive() { return context_.reorder_prim; } in GetPrimitive() function in tensorflow::MklReorderWithScalePrimitive |
| D | mkl_slice_op.cc | 216 std::shared_ptr<primitive> GetPrimitive() { return context_.reorder_prim; } in GetPrimitive() function in tensorflow::MklSlicePrimitive |