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

  /external/tensorflow/tensorflow/core/kernels/
mkl_relu_op.cc 581 TensorShape tf_shape_diff_src; local
590 tf_shape_diff_src.AddDim(diff_src_pd.get_size() / sizeof(T));
593 tf_shape_diff_src = src_tensor.shape();
596 tf_shape_diff_src, dnn_shape_diff_src);
    [all...]
mkl_fused_batch_norm_op.cc 1213 TensorShape tf_shape_diff_src; variable
    [all...]

Completed in 51 milliseconds