Searched refs:tf_shape (Results 1 – 2 of 2) sorted by relevance
998 const TensorShape& tf_shape, in AllocateOutputSetMklShape() argument1005 tf_shape, output)); in AllocateOutputSetMklShape()1019 const TensorShape& tf_shape, in AllocateOutputSetMklShape() argument1026 tf_shape, output)); in AllocateOutputSetMklShape()1042 TensorShape tf_shape; in AllocTmpBuffer() local1044 tf_shape.AddDim(pd.get_size() / sizeof(T) + 1); in AllocTmpBuffer()1046 tf_shape, tensor_out)); in AllocTmpBuffer()1052 TensorShape tf_shape; in AllocTmpBuffer() local1054 tf_shape.AddDim( in AllocTmpBuffer()1059 tf_shape, tensor_out)); in AllocTmpBuffer()[all …]
1072 tf_shape = [None, 5, 4]1073 inp = array_ops.placeholder(dtypes.float32, tf_shape)