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

  /external/tensorflow/tensorflow/core/kernels/
mkl_aggregate_ops.cc 392 memory::format src1_mkl_data_format = src1_mkl_shape.GetTfDataFormat();
400 memory::format src2_mkl_data_format = src2_mkl_shape.GetTfDataFormat();
468 src1_mkl_shape.GetTfDataFormat());
472 src2_mkl_shape.GetTfDataFormat());
mkl_input_conversion_op.cc 321 input_shape_0.GetTfDataFormat());
429 mkl_shape->GetTfDataFormat());
mkl_relu_op.cc 432 dnn_shape_src.GetTfDataFormat());
525 memory::format src_mkl_data_format = dnn_shape_src.GetTfDataFormat();
538 dnn_shape_diff_dst.GetTfDataFormat();
589 dnn_shape_src.GetTfDataFormat());
    [all...]
mkl_reshape_op.cc 154 if (mkl_shape_input.GetTfDataFormat() == input_mkl_md.data.format) {
mkl_concat_op.cc 716 auto orig_tf_format = input_shapes[0].GetTfDataFormat();
756 input_shapes[0].GetTfDataFormat());
    [all...]
mkl_lrn_op.cc 820 memory::format input_format = src_dnn_shape.GetTfDataFormat();
    [all...]
  /external/tensorflow/tensorflow/core/util/
mkl_util.h 517 inline memory::format GetTfDataFormat() const {
    [all...]

Completed in 185 milliseconds