Searched refs:ksize_shape (Results 1 – 1 of 1) sorted by relevance
81 const TensorShape ksize_shape = ctx->InputShape(1); in GetKernelSize() local83 if (!TensorShapeUtils::IsVector(ksize_shape)) { in GetKernelSize()85 ksize_shape.DebugString()); in GetKernelSize()87 if (ksize_shape.num_elements() != num_dims()) { in GetKernelSize()291 const TensorShape ksize_shape = ctx->InputShape(3); in Compile() local293 OP_REQUIRES(ctx, TensorShapeUtils::IsVector(ksize_shape), in Compile()295 ksize_shape.DebugString())); in Compile()511 const TensorShape ksize_shape = ctx->InputShape(3); in Compile() local513 OP_REQUIRES(ctx, TensorShapeUtils::IsVector(ksize_shape), in Compile()515 ksize_shape.DebugString())); in Compile()