HomeSort by relevance Sort by last modified time
    Searched refs:dnn_shape_src (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
mkl_relu_op.cc 388 MklDnnShape dnn_shape_src; variable
389 GetMklShape(context, src_index, &dnn_shape_src);
403 if (dnn_shape_src.IsMklTensor()) {
404 src_md = dnn_shape_src.GetMklLayout();
425 if (dnn_shape_src.IsMklTensor()) {
430 dnn_shape_dst.SetTfLayout(dnn_shape_src.GetDimension(),
431 dnn_shape_src.GetSizesAsMklDnnDims(),
432 dnn_shape_src.GetTfDataFormat());
487 MklDnnShape dnn_shape_src, dnn_shape_diff_dst; local
488 GetMklShape(context, src_index, &dnn_shape_src);
643 MklDnnShape dnn_shape_src; local
706 MklDnnShape dnn_shape_src; local
778 MklDnnShape dnn_shape_src; local
    [all...]
mkl_fused_batch_norm_op.cc 719 MklDnnShape dnn_shape_src; variable
1061 MklDnnShape dnn_shape_src, dnn_shape_diff_dst; variable
    [all...]

Completed in 69 milliseconds