Home
last modified time | relevance | path

Searched defs:input_shape_1 (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_input_conversion_op.cc62 MklDnnShape input_shape_1; in Compute() local
/external/pytorch/torch/csrc/jit/passes/onnx/
Dshape_type_inference.cpp771 auto input_shape_1 = ConstantValueMap::GetShape(n->input(1)->debugName()); in ProcessBroadcastNode() local
876 auto input_shape_1 = in ProcessMatMulNode() local