Home
last modified time | relevance | path

Searched defs:input_tf_shape (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_conv_ops.h549 auto input_tf_shape = GetTfShape(context_, src_index); in GetOutputAndPadSizeInMklOrder() local
H A Dmkl_slice_op.cc85 TensorShape input_tf_shape = input_mkl_shape.IsMklTensor() in ValidateMklInputs() local
H A Dmkl_conv_grad_input_ops.cc527 TensorShape input_tf_shape; in MakeInputTfShape() local