Home
last modified time | relevance | path

Searched refs:native_format (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_grad_input_ops.cc66 bool native_format; member
73 MklTensorFormat tf_fmt, bool native_format, in MklConvBwdInputParams()
81 native_format(native_format), in MklConvBwdInputParams()
183 if (convBwdInputDims.native_format) { in Setup()
285 if (convBwdInputDims.native_format) { in CreateKey()
303 template <typename Device, class T, bool is_depthwise, bool native_format>
320 GetMklShape(context, kInputIdx, &src_mkl_shape, native_format); in Compute()
321 GetMklShape(context, kFilterIdx, &filter_mkl_shape, native_format); in Compute()
322 GetMklShape(context, kOutbpropIdx, &diff_dst_mkl_shape, native_format); in Compute()
343 GetTfShape(context, kOutbpropIdx, native_format); in Compute()
[all …]
Dmkl_matmul_op_fused.cc30 template <typename Device, typename T, bool native_format = false>
65 GetMklShape(ctx, this->kInputIndexSrc, &src_mkl_shape, native_format); in Compute()
66 GetMklShape(ctx, this->kInputIndexWeight, &weight_mkl_shape, native_format); in Compute()
145 GetMklShape(ctx, kInputIndex_Add, &add_mkl_shape, native_format); in Compute()
148 if (native_format && ctx->forward_input_to_output_with_shape( in Compute()
152 } else if (!native_format && ForwardMklTensorInToOutWithMklShape( in Compute()
161 native_format); in Compute()
175 if (native_format) { in Compute()
193 output_mkl_shape, native_format); in Compute()
Dmkl_conv_grad_filter_ops.cc62 bool native_format; member
70 MklTensorFormat tf_fmt, bool native_format, in MklConvBwdFilterParams()
79 native_format(native_format), in MklConvBwdFilterParams()
206 if (convBwdFilterDims.native_format) { in Setup()
340 if (convBwdFilterDims.native_format) { in CreateKey()
360 bool native_format>
377 GetMklShape(context, kInputIdx, &src_mkl_shape, native_format); in Compute()
378 GetMklShape(context, kFilterIdx, &filter_mkl_shape, native_format); in Compute()
379 GetMklShape(context, kDiffDstIdx, &diff_dst_mkl_shape, native_format); in Compute()
391 GetTfShape(context, kDiffDstIdx, native_format); in Compute()
[all …]
Dmkl_fused_batch_norm_op.cc638 bool is_batch_norm_ex = false, bool native_format = false>
693 GetMklShape(context, kSrcIndex, &dnn_shape_src, native_format); in Compute()
837 if (!native_format && src_md != bn_fwd_pd->src_desc()) { in Compute()
856 if (native_format) { in Compute()
860 dnn_shape_dst, native_format); in Compute()
948 dnn_shape_dst, native_format); in HandleEmptyInput()
987 native_format); in AllocateTFOutputs()
1000 mkl_shape_batch_variance, native_format); in AllocateTFOutputs()
1012 native_format); in AllocateTFOutputs()
1023 mkl_shape_saved_variance, native_format); in AllocateTFOutputs()
[all …]
Dmkl_conv_ops.cc68 bool native_format; member
82 MklTensorFormat tf_fmt, bool native_format) in MklConvFwdParams()
92 native_format(native_format) {} in MklConvFwdParams()
211 if (convFwdDims.native_format) { in Setup()
374 if (convFwdDims.native_format) { in CreateKey()
416 bool native_format>
503 GetMklShape(context, kInputIndex_Src, &src_mkl_shape, native_format); in Compute()
505 native_format); in Compute()
537 auto src_tf_shape = GetTfShape(context, kInputIndex_Src, native_format); in Compute()
539 GetTfShape(context, kInputIndex_Filter, native_format); in Compute()
[all …]
Dmkl_avgpooling_op.cc39 template <typename Device, typename T, bool native_format = false>
47 this->native_format_ = native_format; in MklAvgPoolingOp()
169 template <class Device, class T, bool native_format = false>
174 this->native_format_ = native_format; in MklAvgPoolingGradOp()
Dmkl_maxpooling_op.cc43 template <typename Device, typename T, bool native_format = false>
51 this->native_format_ = native_format; in MklMaxPoolingOp()
228 template <class Device, class T, bool native_format = false>
233 this->native_format_ = native_format; in MklMaxPoolingGradOp()
Dmkl_pooling_ops_common.h51 bool native_format; member
58 bool native_format) in MklPoolingParams()
69 native_format(native_format) {} in MklPoolingParams()
Dmkl_pooling_ops_common.cc48 fwdParams.native_format in Setup()
142 bwdParams.native_format in Setup()
/external/libcxx/test/std/input.output/filesystems/fs.enum/
Denum.path.format.pass.cpp34 E::auto_format != E::native_format && in main()
36 E::native_format != E::generic_format, in main()
/external/llvm-project/libcxx/test/std/input.output/filesystems/fs.enum/
Denum.path.format.pass.cpp33 E::auto_format != E::native_format && in main()
35 E::native_format != E::generic_format, in main()
/external/libcxx/test/std/input.output/filesystems/class.path/path.member/path.construct/
Dsource.pass.cpp86 RunTestCaseImpl<CharT>(MS, fs::path::format::native_format); in RunTestCase()
/external/llvm-project/libcxx/test/std/input.output/filesystems/class.path/path.member/path.construct/
Dsource.pass.cpp88 RunTestCaseImpl<CharT>(MS, fs::path::format::native_format); in RunTestCase()
/external/kernel-headers/original/uapi/linux/
Dfd.h182 int native_format; /* native format of this drive */ member
/external/mesa3d/src/gallium/auxiliary/util/
Du_vbuf.c110 enum pipe_format native_format[PIPE_MAX_ATTRIBS]; member
677 enum pipe_format output_format = mgr->ve->native_format[i]; in u_vbuf_translate_begin()
820 ve->native_format[i] = format; in u_vbuf_create_vertex_elements()
822 util_format_get_blocksize(ve->native_format[i]); in u_vbuf_create_vertex_elements()
/external/libcxx/include/
Dfilesystem756 native_format,
/external/llvm-project/libcxx/include/
Dfilesystem773 native_format,